Pytorch - Caught StopIteration in replica 1 on device 1 error while Training on GPU

Question

I am trying to train a BertPunc model on the train2012 data used in the git link: https://github.com/nkrnrnk/BertPunc. While running on the server, with 4 GPUs enabled, belo