colab模型路径保存问题

[pxlsdz]

cloab 运行显示保存模型路径为None，请问我应该在哪里设置？

[CheaSim]

模型保存路径是根据ModelCheckpoint这个回调类来定义的，

model_checkpoint = pl.callbacks.ModelCheckpoint(monitor="Eval/f1", mode="max",
    filename='{epoch}-{Eval/f1:.2f}',
    dirpath="output",
    save_weights_only=True
)

请确保你定义了正确的保存指标，比如Eval/f1，之后设置了dirpath和filename。 如果你想使用transformers的模型保存函数save_pretrained，可以在最后加上

litmodel.model.save_pretrained(path)
tokenizer.save_pretrained(path)