search

znxlwm / pytorch-CartoonGAN

Pytorch implementation of CartoonGAN (CVPR 2018)
MIT License
386 stars 74 forks source link

a question on RuntimeError: CUDA error: out of memory #1

Closed arf111 closed 6 years ago

arf111 commented 6 years ago

hello there, I am currently facing an issue of cuda out of memory for running this code. I have Nvidia 1060 6GB. Is there any way to solve this issue?

znxlwm commented 6 years ago

Hi, The default batch size for this repo is 8. Reduce the batch size to fit your GPU.

python CartoonGAN.py --name your_project_name --src_data src_data_path --tgt_data tgt_data_path --vgg_model pre_trained_VGG19_model_path --batch_size batch_size

arf111 commented 6 years ago

Hi, The default batch size for this repo is 8. Reduce the batch size to fit your GPU.

python CartoonGAN.py --name your_project_name --src_data src_data_path --tgt_data tgt_data_path --vgg_model pre_trained_VGG19_model_path --batch_size batch_size

Thank you so much