原文 标签 python pytorch vgg-net. 我通过使用以下语句来保存具有知识转移的VGG16: torch.save(model.state_dict(), 'checkpoint.pth') 并使用以下语句重新加载:
確定! 回上一頁