Skip to content
Snippets Groups Projects
Unverified Commit 08a3adb5 authored by Haian Huang(深度眸)'s avatar Haian Huang(深度眸) Committed by GitHub
Browse files

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

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

* update

* add train
parent 4705e1fe
No related branches found
No related tags found
No related merge requests found
......@@ -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:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment