diff --git a/happypose/pose_estimators/cosypose/cosypose/training/train_detector.py b/happypose/pose_estimators/cosypose/cosypose/training/train_detector.py index 5ec97ee8..8bdedde7 100644 --- a/happypose/pose_estimators/cosypose/cosypose/training/train_detector.py +++ b/happypose/pose_estimators/cosypose/cosypose/training/train_detector.py @@ -322,9 +322,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)