Merge branch 'training_cosypose' of github.com:agimus-project/happypo…

…se into training_cosypose
agimus-project · Mar 25, 2024 · 91ee620 · 91ee620
2 parents 4bf9b57 + f5829a0
commit 91ee620
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/happypose/pose_estimators/cosypose/cosypose/training/train_detector.py b/happypose/pose_estimators/cosypose/cosypose/training/train_detector.py
@@ -319,9 +319,7 @@ def train_epoch():
                     max_norm=np.inf,
                     norm_type=2,
                 )
-                meters_train["grad_norm"].add(
-                    torch.as_tensor(total_grad_norm).item()
-                )
+                meters_train["grad_norm"].add(torch.as_tensor(total_grad_norm).item())
 
                 optimizer.step()
                 meters_time["backward"].add(time.time() - t)