提交 340b5e80 编写于 作者: S ScXfjiang

fix window_size and sampler

上级 bf11f905
......@@ -162,7 +162,6 @@ def make_data_loader(cfg, is_train=True, is_distributed=False, start_iter=0):
sampler = torch.utils.data.sampler.SequentialSampler(dataset)
else:
sampler = make_data_sampler(dataset, shuffle, is_distributed)
sampler = torch.utils.data.sampler.SequentialSampler(dataset)
batch_sampler = make_batch_data_sampler(
dataset, sampler, aspect_grouping, images_per_gpu, num_iters, start_iter
)
......
......@@ -10,7 +10,7 @@ class SmoothedValue(object):
window or the global series average.
"""
def __init__(self, window_size=20):
def __init__(self, window_size=1):
self.deque = deque(maxlen=window_size)
self.series = []
self.total = 0.0
......@@ -61,6 +61,6 @@ class MetricLogger(object):
loss_str = []
for name, meter in self.meters.items():
loss_str.append(
"{}: {:.4f} ({:.4f})".format(name, meter.median, meter.global_avg)
"{}: {:.4f} ({:.4f})".format(name, meter.median, meter.avg)
)
return self.delimiter.join(loss_str)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册