yjh0410 1 rok temu
rodzic
commit
27c15f16c2
1 zmienionych plików z 1 dodań i 0 usunięć
  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