如何在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

本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
THE END
分享
二维码
< <上一篇
下一篇>>