摘要：前言 PyTorch 提供了一种非常方便的节省显存的方式，就是 Checkpoint 机制。这篇文章的目的在于更透彻的了解其内在机制。 Checkpoint 机制 该技术的核心是一种使用时间换空间的策略。在现有的许多方法中被大量使用，例如 DenseNet、Swin Transformer 源码中都 阅读全文

摘要：open cmd `nvidia-smi` `taskkill -PID "pidNumber" -F ` 阅读全文

摘要：t = torch.arange(72).reshape(2,3,3,4) # 初始化一个tensor print("t >", t) index = torch.tensor([1, 0]) # 要选取数据的位置 print("index >", index) data0 = t.index_se 阅读全文