diff --git a/happypose/pose_estimators/cosypose/cosypose/training/train_detector.py b/happypose/pose_estimators/cosypose/cosypose/training/train_detector.py index fdb4675f..4ced45bd 100644 --- 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)