报错如下:
Traceback (most recent call last):
File "./train.py", line 348, in train
inputs_x, targets_x = labeled_iter.next()
File "/home/user-wsc/anaconda3/envs/yjhtorch16/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 652, in __next__
data = self._next_data()
File "/home/user-wsc/anaconda3/envs/yjhtorch16/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1320, in _next_data
raise StopIteration
StopIteration
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File "./train.py", line 536, in <module>
main()
File "./train.py", line 314, in main
train(args, labeled_trainloader, unlabeled_trainloader, test_loader,
File "./train.py", line 356, in train
inputs_x, targets_x = labeled_iter.next()
File "/home/user-wsc/anaconda3/envs/yjhtorch16/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 652, in __next__
data = self._next_data()
File "/home/user-wsc/anaconda3/envs/yjhtorch16/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1320, in _next_data
raise StopIteration
数据加载的时候报错,发现是因为batch_size的大小大于我的样本数量了,将batch_size改小即可。