batch_iterator = iter(data_loader)
File "D:\ProgramData\Anaconda3\envs\mytorch\lib\site-packages\torch\utils\data\dataloader.py", line 368, in __iter__
return self._get_iterator()
File "D:\ProgramData\Anaconda3\envs\mytorch\lib\site-packages\torch\utils\data\dataloader.py", line 314, in _get_iterator
return _MultiProcessingDataLoaderIter(self)
File "D:\ProgramData\Anaconda3\envs\mytorch\lib\site-packages\torch\utils\data\dataloader.py", line 965, in __init__
self._reset(loader, first_iter=True)
File "D:\ProgramData\Anaconda3\envs\mytorch\lib\site-packages\torch\utils\data\dataloader.py", line 996, in _reset
self._try_put_index()
File "D:\ProgramData\Anaconda3\envs\mytorch\lib\site-packages\torch\utils\data\dataloader.py", line 1230, in _try_put_index
index = self._next_index()
File "D:\ProgramData\Anaconda3\envs\mytorch\lib\site-packages\torch\utils\data\dataloader.py", line 521, in _next_index
return next(self._sampler_iter) # may raise StopIteration
File "D:\ProgramData\Anaconda3\envs\mytorch\lib\site-packages\torch\utils\data\sampler.py", line 226, in __iter__
for idx in self.sampler:
File "D:\ProgramData\Anaconda3\envs\mytorch\lib\site-packages\torch\utils\data\sampler.py", line 122, in __iter__
yield from torch.randperm(n, generator=generator).tolist()
RuntimeError: Expected a 'cuda' device type for generator but found 'cpu'
将
data_loader = data.DataLoader(dataset, args.batch_size,
num_workers=args.num_workers,
shuffle=True, collate_fn=detection_collate,
pin_memory=True)
改为
data_loader = data.DataLoader(dataset, args.batch_size,
num_workers=args.num_workers,
shuffle=True, collate_fn=detection_collate,
pin_memory=True,generator=torch.Generator(device = 'cuda'))