训练FFNN pyTorch时记忆力不断提高我有一个前馈神经网络来分类MNIST数据集。 由于某些原因,不管批大小有多大,内存都会保持在99%左右。 我没有任何在大小上增加的东西-每个动态变量在第一个epoch之后都会被重写,但即使在epoch ...2024-10-01 已阅读: n次