摘要:
1 #coding= utf-8 2 import os 3 import torch 4 from data_pipe import get_data 5 from model import SimpleNet 6 import numpy as np 7 import cv2 8 from PI 阅读全文
摘要:
模型保存与加载有两种方式,本文暂时只讨论模型参数方式 1> 单GPU 保存 1 torch.save(model.state_dict(), "model.pth") 加载 1 model = SimpleNet() 2 model.load_state_dict(torch.load("./mod 阅读全文