collie.utils.find_tensors¶
- find_tensors()[源代码]¶
打印出垃圾回收区的所有张量。
Adopted from https://discuss.pytorch.org/t/how-to-debug-causes-of-gpu-memory-leaks/6741/3
打印出垃圾回收区的所有张量。
Adopted from https://discuss.pytorch.org/t/how-to-debug-causes-of-gpu-memory-leaks/6741/3