Out of memory - 1060 6GB while Sampling

taoxugit / AttnGAN

MIT License

1.33k stars 415 forks source link

Out of memory - 1060 6GB while Sampling #22

Open saxenarohit opened 5 years ago

saxenarohit commented 5 years ago

RuntimeError: CUDA error: out of memory I am running sampling for birds model (python main.py --cfg cfg/eval_bird.yml --gpu 0)

nitish11 commented 5 years ago

Change the batch size defined in cfg/eval_bird.yml.

Updated parameters :

TRAIN:
    FLAG: True
    BATCH_SIZE: 16

ahaliu1 commented 4 years ago

Change the batch size defined in cfg/eval_bird.yml.

Updated parameters :
TRAIN:
    FLAG: True
    BATCH_SIZE: 16
should I set flag true? Does it means I need to retrain the nerwork? but I just need to running a sample

nn970510 commented 3 years ago

Change the batch size defined in cfg/eval_bird.yml. Updated parameters :
TRAIN:
    FLAG: True
    BATCH_SIZE: 16
should I set flag true? Does it means I need to retrain the nerwork? but I just need to running a sample

Should I set it as true? Same question

nn970510 commented 3 years ago

RuntimeError: CUDA error: out of memory I am running sampling for birds model (python main.py --cfg cfg/eval_bird.yml --gpu 0)

Did you solve it?