如何才能使用内存小或者显存小的设备训练神经网络

每一轮就清空一次，加载新的数据集 每轮保存一次模型，每轮都加载预训练模型 或者在每次输入数据前使用 torch.cuda.empty_cache()这句话及时的清理显存 python 深度学习与PyTorch实战web for epoch in range(60): torch.cuda.empty_cache() if epoch:

>>阅读原文<<