pytorch中load和load_state_dict区别

torch.load 加载的是训练好的模型 例如python torch.load("ad.pth") 而load_state_dict是net的一个方法 是将torch.load加载出来的数据加载到net中web
相关文章
相关标签/搜索