diff --git a/mmcv/runner/runner.py b/mmcv/runner/runner.py index 6eb932521..d90275066 100644 --- a/mmcv/runner/runner.py +++ b/mmcv/runner/runner.py @@ -262,9 +262,6 @@ class Runner(object): self.call_hook('after_train_iter') self._iter += 1 - print('**max cached: {}\n'.format(torch.cuda.max_memory_cached())) - print('**max allocated: {}\n'.format(torch.cuda.max_memory_allocated())) - self.call_hook('after_train_epoch') self._epoch += 1