iSeeBetter
iSeeBetter copied to clipboard
Problem when testing
I have a problem when testing. I run python3 iSeeBetterTest.py --output Result -c --data_dir Vid4 --file_list foliage_test.txt -m weights/netG_epoch_4_1.pth --upscale_only
my parameters are
(chop_forward=False, data_dir='Vid4', debug=False, file_list='foliage_test.txt', future_frame=True, gpu_mode=True, gpus=1, model='weights/netG_epoch_4_ 1.pth', model_type='RBPN', nFrames=7, other_dataset=True, output='Result', residual=False, seed=123, testBatchSize=1, threads=0, upscale_factor=4, upscale_only =True)
and finally I get an CUDA memory error.. Please someone help me. Thank you