microsoft / Bringing-Old-Photos-Back-to-Life

Bringing Old Photo Back to Life (CVPR 2020 oral)
https://arxiv.org/abs/2004.09484
MIT License
14.94k stars 1.97k forks source link

AssertionError: Batch size 4 is wrong. It must be a multiple of # GPUs 3. #203

Open davaram opened 2 years ago

davaram commented 2 years ago

python3.6 run.py --input_folder test_images/old_w_scratch_gpu012/ --output_folder old_w_scratch_output-gpu-012/ --GPU 0,1,2

Error comming... help please

File "test_face.py", line 15, in opt = TestOptions().parse() File "/home/ubuntu/download/Bringing-Old-Photos-Back-to-Life/Face_Enhancement/options/base_options.py", line 293, in parse ), "Batch size %d is wrong. It must be a multiple of # GPUs %d." % (opt.batchSize, len(opt.gpu_ids)) AssertionError: Batch size 4 is wrong. It must be a multiple of # GPUs 3. Finish Stage 3 ...

davaram commented 2 years ago

so , what is batchSize?