StopIteration: Caught StopIteration in replica 0 on device 0
在对调用pytorch_pretrained_bert时,如果用多个GPU出现StopIteration: Caught StopIteration in replica 0 on device 0.具体如下。
File "/home/yuangen_yu/CLUE/baselines/models_pytorch/classifier_pytorch/run_classifier.py", line 569, in
main()
File "/home/yuangen_yu/CLUE/baselines/
原创
2020-10-17 01:06:26 ·
4913 阅读 ·
8 评论