摘要: 1.*.pt文件 .pt文件保存的是模型的全部,在加载时可以直接赋值给新变量model = torch.load("filename.pt")。 具体操作: (1). 模型的保存 torch.save(model,"Path/filename.pt") (2). 模型的加载 model = torc 阅读全文
posted @ 2024-07-19 10:09 华小电 阅读(354) 评论(0) 推荐(0) 编辑