fix possible bugs brought by dumping grad_norm to the logger in pytorch1.5 (#349)

3f4c9fd0 · Wang Xinjiang · GitHub · f28a7c7e · 3f4c9fd0
Unverified Commit 3f4c9fd0 authored Jun 16, 2020 by Wang Xinjiang Committed by GitHub Jun 16, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

mmcv/runner/hooks/optimizer.py mmcv/runner/hooks/optimizer.py +1 -1

No files found.
--- a/mmcv/runner/hooks/optimizer.py
+++ b/mmcv/runner/hooks/optimizer.py
@@ -23,6 +23,6 @@ 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': grad_norm},
+                runner.log_buffer.update({'grad_norm': float(grad_norm)},
                                         runner.outputs['num_samples'])
        runner.optimizer.step()