ShuaiBai623
commented
2 years ago try add ”drop_last=True“ in line 45, "test_loader = data.DataLoader(test_dataset, batch_size=opt.batch_size, shuffle=opt.shuffle,num_workers=opt.workers)"
Closed maizer2 closed 1 year ago
ShuaiBai623
commented
2 years ago try add ”drop_last=True“ in line 45, "test_loader = data.DataLoader(test_dataset, batch_size=opt.batch_size, shuffle=opt.shuffle,num_workers=opt.workers)"
python -u test_SDAFNet_viton.py -b 8 --name TEST_PAIR --mode test --dataset_list VITON/test_pairs.txt --workers 0