ludwig 覆盖预训练模型文件

flmtquvp  于 2个月前  发布在  其他
关注(0)|答案(1)|浏览(27)

这可能是一个与#1028相关的功能请求。我相信在加载预训练模型时覆盖所有模型参数是可以改进的。以下是情景:
运行1:运行100个epoch,将early_stop设置为10个epoch。现在模型文件已准备好。
运行2:新数据和新模型文件,但模型中唯一的变化是将early_stop设置为2个epoch(例如)。当前实现只会简单地覆盖训练参数,并在达到100个epoch或early stop值为10时停止运行。
允许修改一些参数,如epoch数量或early stop值,将在后续运行中节省时间。我现在正在修改从运行1开始的model_hyperparameters.json文件以实现相同的效果。当使用慢速学习率时,每个epoch可能需要很长时间-因此后续运行中的较早停止值可能会有所帮助。

相关问题