From f5829a0a72159260c19b189dc118e70b55a8278c Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Mon, 25 Mar 2024 12:27:36 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- .../cosypose/cosypose/training/train_detector.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) 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)