yjh0410 1 年之前
父節點
當前提交
27c15f16c2
共有 1 個文件被更改,包括 1 次插入0 次删除
  1. 1 0
      engine.py

+ 1 - 0
engine.py

@@ -930,6 +930,7 @@ class RTCTrainer(object):
         metric_logger = MetricLogger(delimiter="  ")
         metric_logger.add_meter('lr', SmoothedValue(window_size=1, fmt='{value:.6f}'))
         metric_logger.add_meter('size', SmoothedValue(window_size=1, fmt='{value:d}'))
+        metric_logger.add_meter('grad_norm', SmoothedValue(window_size=1, fmt='{value:.1f}'))
         header = 'Epoch: [{} / {}]'.format(self.epoch, self.args.max_epoch)
         epoch_size = len(self.train_loader)
         print_freq = 10