How to reset MaxBytesInUse in Tensorflow 1.13

后端未结

关注

 0  870

I want to measure peak memory usage in TensorFlow 1.13 per epoch.

In PyTorch there is a very simple solution by using combination of two functions: torch.cuda.m


                      
              相关标签: