yjh0410 1 년 전
부모
커밋
887564dc6c
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      yolo/engine.py

+ 1 - 1
yolo/engine.py

@@ -214,7 +214,7 @@ class YoloTrainer(object):
 
             # Backward
             self.scaler.scale(losses).backward()
-            gnorm = get_total_grad_norm(model.parameters())
+            gnorm = 0.0
 
             # Optimize
             if (iter_i + 1) % self.grad_accumulate == 0: