Commit de70fe5c authored by pangjm's avatar pangjm
Browse files

rm useless

parent ecce8572
...@@ -262,9 +262,6 @@ class Runner(object): ...@@ -262,9 +262,6 @@ class Runner(object):
self.call_hook('after_train_iter') self.call_hook('after_train_iter')
self._iter += 1 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.call_hook('after_train_epoch')
self._epoch += 1 self._epoch += 1
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment