AttnGAN icon indicating copy to clipboard operation
AttnGAN copied to clipboard

Published 20 hours ago • verified publisher icon taoxugit

Out of memory - 1060 6GB while Sampling

Open saxenarohit opened this issue 6 years ago • 4 comments

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

saxenarohit avatar Aug 07 '18 13:08 saxenarohit

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

Updated parameters :

TRAIN:
    FLAG: True
    BATCH_SIZE: 16

nitish11 avatar Sep 17 '18 07:09 nitish11

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

ahaliu1 avatar Jul 10 '19 02:07 ahaliu1

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 avatar Mar 28 '21 21:03 nn970510

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?

nn970510 avatar Mar 28 '21 21:03 nn970510