AttnGAN
AttnGAN copied to clipboard
Training on CPU
I have no GPU, could I use my CPU for the training? Thank you very much
If you don't pass a GPU parameter while calling pretrain_DAMSM or main.py, it should default to CPU. Refer to #23 to fix trainer.py