Inference for swap.py error

YuvalNirkin / fsgan

FSGAN - Official PyTorch Implementation

https://nirkin.com/fsgan

Creative Commons Zero v1.0 Universal

754 stars 147 forks source link

Closed tushar-31093 closed 3 years ago

tushar-31093 commented 3 years ago

Can anyone suggest a solution to this? Windows 10, Conda, Pytorch with CUDA, Batch size 1, Test image to target video

error

tushar-31093 commented 3 years ago

After adding : python swap.py input.mp4 -t 3.mp4 -o res.mp4 --seg_remove_mouth --encoder_codec mp4v -db 1 -pb 1 -lb 1 -sb 1

This error is coming now.

error2

YuvalNirkin commented 3 years ago

4GB GPU is not enough for running the method. At least 8GB should be enough.