out of memory

[zxt-triumph]

when I run command that bash ./scripts/train_edges2shoes.sh
I got following error
RuntimeError: cuda runtime error (2) : out of memory at /opt/conda/conda-bld/pytorch_1518243271935/work/torch/lib/THC/generic/THCStorage.cu:58
how to solve this problem?

[junyanz]

how big is your gpu memory? You may want to use smaller images and smaller models. ( --loadSize 143 --fineSize 128 --which_model_netD basic_128_multi --which_model_netD2 basic_128_multi --which_model_netG unet_128 --which_model_netE resnet_128)

[zxt-triumph]

Thank you for your reply.My gpu memory is only 3G.According to your suggestion, I modified the memory but the problem is still not resolved.

[junyanz]

Could you give more details about your current problem?

[junyanz]

These warnings are related to visdom. Try python -m visdom.server. See this post for more details.

[zxt-triumph]

Thank you so much.I have solved this problem with your help.