Unable to reproduce the generated image
Opened this issue · 4 comments
ryhhtn commented
ryhhtn commented
I'm using following library
torch==1.7.1+cu92
torchfile==0.1.0
torchinfo==0.0.9
torchmetrics==0.3.1
torchtext==0.8.1
torchvision==0.8.2+cu92
kobrien-Novetta commented
kobrien-Novetta commented
Looking at issue #6, apparently it only works on 2 gpus. If you change the gpu input to 0,0 you'll be able to run it on 1 gpu! If you can't pass 0,0 into on the command line, then you can just edit the default paramter for gpu_ids to be '0,0'