AttnGAN icon indicating copy to clipboard operation
AttnGAN copied to clipboard

Training on CPU

Open albertocattani opened this issue 6 years ago • 1 comments

I have no GPU, could I use my CPU for the training? Thank you very much

albertocattani avatar May 21 '18 17:05 albertocattani

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

shaivaldalal avatar Dec 04 '18 14:12 shaivaldalal