摘要: os模块是python标准库中的一个用于访问操作系统功能的模块。 系统操作 1 os.sep——获取路径分隔符 在Windows上,文件的路径分隔符是'\',在Linux上是'/'。例如:文件夹img下有一张图lake.jpg, windows的路径:img\lake.jpg Linux的路径:im 阅读全文
posted @ 2020-05-03 16:48 Picassooo 阅读(173) 评论(0) 推荐(0) 编辑
摘要: 保存和加载模型 在PyTorch中使用torch.save来保存模型的结构和参数,有两种保存方式: # 方式一:保存模型的结构信息和参数信息 torch.save(model, './model.pth') # 方式二:仅保存模型的参数信息 torch.save(model.state_dict() 阅读全文
posted @ 2020-05-03 10:12 Picassooo 阅读(446) 评论(0) 推荐(0) 编辑