如何使用使用哈佛pystruct对自定义数据进行训练的模型进行预测？

Question

我在pytorch中是个新手。我正在尝试使用以下资源使用自定义数据训练模型 https://github.com/harvardnlp/pytorch-struct/blob/master/notebooks/BertDependencies.ipynb

我期望torch.save和model.load ....像函数调用一样。

不幸的是，我无法弄清楚代码是如何保存模型的。保存后，如何使用该模型预测带有任何例句的某些任务？请在这里放点灯吗？预先谢谢你。