一文弄懂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)

上一篇 下一篇

猜你喜欢

热点阅读