mirrors/mmengine
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmengine.git synced 2025-06-03 21:54:44 +08:00
Code Issues Projects Releases Wiki Activity

Fix error of 'Runner' object has no attribute 'log_buffer' (#259)

Browse Source 
* fix 'Runner' object has no attribute 'log_buffer'

* update

* add train
This commit is contained in:
Haian Huang(深度眸) 2022-05-27 10:51:25 +08:00 committed by GitHub
parent 4705e1fe3d
commit 08a3adb5d7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mmengine/hooks/optimizer_hook.py
View File

@ -92,8 +92,8 @@ class OptimizerHook(Hook):
grad_norm = self.clip_grads(runner.model.parameters()) grad_norm = self.clip_grads(runner.model.parameters())
if grad_norm is not None: if grad_norm is not None:
# Add grad norm to the logger # Add grad norm to the logger
runner.log_buffer.update({'grad_norm': float(grad_norm)}, runner.message_hub.update_scalar('train/grad_norm',
runner.outputs['num_samples']) float(grad_norm))
runner.optimizer.step() runner.optimizer.step()
def detect_anomalous_parameters(self, loss: torch.Tensor, runner) -> None: def detect_anomalous_parameters(self, loss: torch.Tensor, runner) -> None: