pytorch中load和load_state_dict區別

torch.load 加載的是訓練好的模型 例如python torch.load("ad.pth") 而load_state_dict是net的一個方法 是將torch.load加載出來的數據加載到net中web
相關文章
相關標籤/搜索