AutoGluon微调Chronos报错

	Warning: Exception caused Chronosfine[..__models__chronos-bolt-small] to fail during training... Skipping this model.
	TrainingArguments.__init__() got an unexpected keyword argument 'evaluation_strategy'

这个提示出现在微调的时候,在zero-shot的时候没有发现。

predictor = TimeSeriesPredictor(
    prediction_length=int(prediction_length),
    target="load",
).fit(
    train_data=train_data,
    hyperparameters={
        "Chronos": [
            {
                "model_path": "..\models\chronos-bolt-small",
                "fine_tune_lr": 1e-4,
                "ag_args": {"name_suffix": "fine"},
                "fine_tune": True,
                "fine_tune_steps": 2000,
                "context_length": 1024
            },
        ],
    },
    enable_ensemble=False,
    time_limit=600,
)

这个原因是由于transformers的版本4.51已经没有这个参数了。下调版本为4.48

pip install transformers==4.48.0
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值