load

mac2024-05-26  26

torch.nn.Module.load_state_dict:https://blog.csdn.net/qq_35608277/article/details/85837457

 

简单的模型load: 一般来说,保存模型是把参数全部用 model.cpu().state_dict(), 然后加载模型时一般用 model.load_state_dict(torch.load(model_path))。 值得注意的是:torch.load 返回的是一个 OrderedDicthttps://blog.csdn.net/hungryof/article/details/81364487

最新回复(0)