提交 280dd0e3 编写于 作者: L LielinJiang

remove self.sample_iter

上级 368d6302
......@@ -62,7 +62,6 @@ class DistributedBatchSampler(BatchSampler):
def __init__(self, dataset, batch_size, shuffle=False, drop_last=False):
self.dataset = dataset
self.sample_iter = None
assert isinstance(batch_size, int) and batch_size > 0, \
"batch_size should be a positive integer"
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册