[Error] Problem running main.py , enquiry regarding hardware requirements

[Acejoy]

Hello @soyeonm , I am running main.py commands given in the README.md , I am getting 'Insufficient GPU Memory'. I have a 4GB Nvidia 1650Ti.

1. I wanted to know the hardware requirements to run this command.
2. Also, is there any way to reduce this GPU memory requirement, as I just want to run the script to see a demo of the learnt models(not train).
3. Also, what would be the GPU requirements for training.

Thanks