如何才能使用內存小或者顯存小的設備訓練神經網絡

每一輪就清空一次，加載新的數據集 每輪保存一次模型，每輪都加載預訓練模型 或者在每次輸入數據前使用 torch.cuda.empty_cache()這句話及時的清理顯存 python 深度學習與PyTorch實戰web for epoch in range(60): torch.cuda.empty_cache() if epoch:

>>阅读原文<<