Pytorch 0.4 DataLoaderIter() based_seed error
Pytorch 0.4 版本在训练中,固定样本,固定 random seed,多次训练结果无法完全复现问题;
有一处 0.4 版本源码问题需要修改,如下:
torch/utils/data/dataloader.py
- base_seed = torch.LongTensor(1).random_()[0]
+ base_seed = torch.LongTensor(1).random_().item()
参考链接 discussion
参考链接 source code