我已经在jupyter notebook中使用vs-code训练了一个大型数据集.如何保存训练好的数据集?

Question

这是关于训练数据集的信息

我花了24小时来训练整个数据集。我猜它已经完全训练了，但它仍然在显示(1150/1583)。

然而，我怎样才能保存我的进度，使我不必重新训练

我试着保存一次代码，但是模型又开始训练了。

## Save Model
torch.save(netG, 'C:/6th sem/TT(L)/Face detection/imgGen.pt')
torch.save(netD, 'C:/6th sem/TT(L)/Face detection/imgDis.pt')

所以我用了这个，我不确定它是否会工作。

Answer 1

您可以使用joblib来保存和加载模型。

from sklearn.externals import joblib

使用joblib.dump()和joblib.load()方法