我在编写Python代码时,想使用Metal来训练我的模型,像这样:
from metal.label_model import LabelModelgen_model = LabelModel()%time gen_model.train(L_train[0], n_epochs=500, print_every=100)
但这会导致:
TypeError Traceback (most recent call last)<timed eval> in <module>TypeError: train() got an unexpected keyword argument 'n_epochs'
回答:
在版本0.3.0中有一个变化:
‘将Classifier.train重命名为Classifier.train_model,以避免覆盖nn.Module.train函数’
因此,尝试使用train_model()代替train():
from metal.label_model import LabelModelgen_model = LabelModel()%time gen_model.train_model(L_train[0], n_epochs=500, print_every=100)
来源:
https://github.com/HazyResearch/metal/commit/4210c7c66f3f4a6fc7287192aec133c293ed8198