mirror of https://github.com/open-mmlab/mmcv.git
rm useless
parent
ecce857270
commit
de70fe5cb0
|
@ -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
|
||||
|
||||
|
|
Loading…
Reference in New Issue