一文弄懂Pytorch的DataLoader, DataSet,
2021-01-25 本文已影响0人
求索_700e
1)https://www.cnblogs.com/marsggbo/p/11308889.html
2)https://zhuanlan.zhihu.com/p/100280685?utm_source=qq
3)为什么要设置set_epoch:见https://www.zhihu.com/question/67209417/answer/1017851899,目的是:
# 设置当前的 epoch,为了让不同的结点之间保持同步。
train_sampler.set_epoch(epoch)