From 08a3adb5d74e3821fec38f00e2701777c2862191 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Haian=20Huang=28=E6=B7=B1=E5=BA=A6=E7=9C=B8=29?=
<1286304229@qq.com>
Date: Fri, 27 May 2022 10:51:25 +0800
Subject: [PATCH] Fix error of 'Runner' object has no attribute 'log_buffer'
(#259)
* fix 'Runner' object has no attribute 'log_buffer'
* update
* add train
---
mmengine/hooks/optimizer_hook.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/mmengine/hooks/optimizer_hook.py b/mmengine/hooks/optimizer_hook.py
index 9107dbf0..6a8f7c11 100644
--- a/mmengine/hooks/optimizer_hook.py
+++ b/mmengine/hooks/optimizer_hook.py
@@ -92,8 +92,8 @@ class OptimizerHook(Hook):
grad_norm = self.clip_grads(runner.model.parameters())
if grad_norm is not None:
# Add grad norm to the logger
- runner.log_buffer.update({'grad_norm': float(grad_norm)},
- runner.outputs['num_samples'])
+ runner.message_hub.update_scalar('train/grad_norm',
+ float(grad_norm))
runner.optimizer.step()
def detect_anomalous_parameters(self, loss: torch.Tensor, runner) -> None:
--
GitLab