如何在kaggle上用pytorch保存、加载文件,同时在output上删除已经保存的文件。

 保存:

torch.save(date,path)

其中date为保存数据,path为路径加文件名,如:torch.save(date, '/kaggle/working'+".pt")

加载

torch.load('/kaggle/working/07pt')

直接选择路径下的文件名进行load

在output上删除已经存在的pt文件

import shutil
import os

if __name__ == '__main__':
    path = '/kaggle/working'
    if os.path.exists(path):
        shutil.rmtree(path)
        print('删除完成')
    else:
        print('原本为空')

直接在code上进行运行,即可删除所有working下的文件,选择删除其他路径的文件只需要进行改变path

你可能感兴趣的:(深度学习,深度学习,人工智能,神经网络,pytorch)