📌  相关文章
📜  如何在 pytorch 中加载预训练模型 - Python 代码示例

📅  最后修改于: 2022-03-11 14:45:37.349000             🧑  作者: Mango

代码示例1
pytorch_model = MNISTClassifier()
pytorch_model.load_state_dict(torch.load(path))
model.eval()