-
Notifications
You must be signed in to change notification settings - Fork 14
Description
Traceback (most recent call last):
File "src/train1113.py", line 418, in
main()
File "src/train1113.py", line 415, in main
train(0,opt)
File "src/train1113.py", line 395, in train
dataset=dataset)
File "src/train1113.py", line 259, in train_epoch
for iteration, batch in tqdm(enumerate(train_data_loader)):
File "/mnt/data/zhangyufei/Anaconda/anaconda3/envs/SLADD/lib/python3.6/site-packages/tqdm/std.py", line 1195, in iter
for obj in iterable:
File "/mnt/data/zhangyufei/Anaconda/anaconda3/envs/SLADD/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 521, in next
data = self._next_data()
File "/mnt/data/zhangyufei/Anaconda/anaconda3/envs/SLADD/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 1203, in _next_data
return self._process_data(data)
File "/mnt/data/zhangyufei/Anaconda/anaconda3/envs/SLADD/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 1229, in _process_data
data.reraise()
File "/mnt/data/zhangyufei/Anaconda/anaconda3/envs/SLADD/lib/python3.6/site-packages/torch/_utils.py", line 434, in reraise
raise exception
TypeError: Caught TypeError in DataLoader worker process 2.
Original Traceback (most recent call last):
File "/mnt/data/zhangyufei/Anaconda/anaconda3/envs/SLADD/lib/python3.6/site-packages/torch/utils/data/_utils/worker.py", line 287, in _worker_loop
data = fetcher.fetch(index)
File "/mnt/data/zhangyufei/Anaconda/anaconda3/envs/SLADD/lib/python3.6/site-packages/torch/utils/data/_utils/fetch.py", line 52, in fetch
return self.collate_fn(data)
File "/mnt/data/zhangyufei/Anaconda/anaconda3/envs/SLADD/lib/python3.6/site-packages/torch/utils/data/_utils/collate.py", line 84, in default_collate
return [default_collate(samples) for samples in transposed]
File "/mnt/data/zhangyufei/Anaconda/anaconda3/envs/SLADD/lib/python3.6/site-packages/torch/utils/data/_utils/collate.py", line 84, in
return [default_collate(samples) for samples in transposed]
File "/mnt/data/zhangyufei/Anaconda/anaconda3/envs/SLADD/lib/python3.6/site-packages/torch/utils/data/utils/collate.py", line 70, in default_collate
return torch.tensor(batch)
TypeError: can't convert np.ndarray of type numpy.str. The only supported types are: float64, float32, float16, complex64, complex128, int64, int32, int16, int8, uint8, and bool.