2021年9月4日
机器学习-网络模型的保存与使用（pytorch环境）
摘要： 例子 vgg16_false = torchvision.models.vgg16(pretrained=False) # 保存方式1（网络模型） torch.save(vgg16_false,"vgg_method1.pth") # 保存方式2（参数模型） torch.save(vgg16_fal 阅读全文
posted @ 2021-09-04 20:04 EA2218764AB 阅读(76) 评论(0) 推荐(0) 编辑