提交 ee40e1fc 编写于 作者: Y Yang Nie 提交者: Tingquan Gao

bugfix: make the `epoch` in MultiScaleSampler self-incrementing

上级 de4129ba
......@@ -103,6 +103,7 @@ class MultiScaleSampler(Sampler):
random.seed(self.seed)
else:
random.seed(self.epoch)
self.epoch += 1
random.shuffle(self.img_indices)
random.shuffle(self.img_batch_pairs)
indices_rank_i = self.img_indices[self.rank:len(self.img_indices):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册