Start to test %s!
Network [SPADEGenerator] was created. Total number of parameters: 100.5 million. To see the architecture, do print(network).
Traceback (most recent call last):
File "/usr/src/SD-VITON/test_generator.py", line 266, in
main()
File "/usr/src/SD-VITON/test_generator.py", line 260, in main
test(opt, test_loader, board, tocg, generator)
File "/usr/src/SD-VITON/test_generator.py", line 115, in test
for inputs in test_loader.data_loader:
File "/usr/local/lib/python3.10/site-packages/torch/utils/data/dataloader.py", line 630, in next
data = self._next_data()
File "/usr/local/lib/python3.10/site-packages/torch/utils/data/dataloader.py", line 1345, in _next_data
return self._process_data(data)
File "/usr/local/lib/python3.10/site-packages/torch/utils/data/dataloader.py", line 1371, in _process_data
data.reraise()
File "/usr/local/lib/python3.10/site-packages/torch/_utils.py", line 694, in reraise
raise exception
RuntimeError: Caught RuntimeError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/usr/local/lib/python3.10/site-packages/torch/utils/data/_utils/worker.py", line 308, in _worker_loop
data = fetcher.fetch(index)
File "/usr/local/lib/python3.10/site-packages/torch/utils/data/_utils/fetch.py", line 51, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/usr/local/lib/python3.10/site-packages/torch/utils/data/_utils/fetch.py", line 51, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/usr/src/SD-VITON/cp_dataset_test.py", line 164, in getitem
parse_map = parsemap.scatter(0, parse, 1.0)
RuntimeError: Index tensor must have the same number of dimensions as self tensor
python3 test_generator.py --occlusion --test_name test1 --tocg_checkpoint /usr/src/SD-VITON/tocg.pth --gpu_ids 0 --gen_checkpoint /usr/src/SD-VITON/toig.pth --dataroot /usr/src/SD-VITON/data --data_list /usr/src/SD-VITON/data/1.txt --composition_mask
Start to test %s! Network [SPADEGenerator] was created. Total number of parameters: 100.5 million. To see the architecture, do print(network). Traceback (most recent call last): File "/usr/src/SD-VITON/test_generator.py", line 266, in
main()
File "/usr/src/SD-VITON/test_generator.py", line 260, in main
test(opt, test_loader, board, tocg, generator)
File "/usr/src/SD-VITON/test_generator.py", line 115, in test
for inputs in test_loader.data_loader:
File "/usr/local/lib/python3.10/site-packages/torch/utils/data/dataloader.py", line 630, in next
data = self._next_data()
File "/usr/local/lib/python3.10/site-packages/torch/utils/data/dataloader.py", line 1345, in _next_data
return self._process_data(data)
File "/usr/local/lib/python3.10/site-packages/torch/utils/data/dataloader.py", line 1371, in _process_data
data.reraise()
File "/usr/local/lib/python3.10/site-packages/torch/_utils.py", line 694, in reraise
raise exception
RuntimeError: Caught RuntimeError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/usr/local/lib/python3.10/site-packages/torch/utils/data/_utils/worker.py", line 308, in _worker_loop
data = fetcher.fetch(index)
File "/usr/local/lib/python3.10/site-packages/torch/utils/data/_utils/fetch.py", line 51, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/usr/local/lib/python3.10/site-packages/torch/utils/data/_utils/fetch.py", line 51, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/usr/src/SD-VITON/cp_dataset_test.py", line 164, in getitem
parse_map = parsemap.scatter(0, parse, 1.0)
RuntimeError: Index tensor must have the same number of dimensions as self tensor