gpt4 book ai didi

amazon-web-services - 使用内置 sagemaker 算法进行增量学习

转载 作者:行者123 更新时间:2023-11-30 08:56:37 26 4
gpt4 key购买 nike

我正在训练 DeepAR AWS SageMaker 内置算法。借助 sagemaker SDK,我可以使用特定的指定超参数来训练模型:

estimator = sagemaker.estimator.Estimator(
sagemaker_session=sagemaker_session,
image_name=image_name,
role=role,
train_instance_count=1,
train_instance_type='ml.c4.2xlarge',
base_job_name='wfp-deepar',
output_path=join(s3_path, 'output')
)

estimator.set_hyperparameters(**{
'time_freq': 'M',
'epochs': '50',
'mini_batch_size': '96',
'learning_rate': '1E-3',
'context_length': '12',
'dropout_rate': 0,
'prediction_length': '12'
})

estimator.fit(inputs=data_channels, wait=True, job_name='wfp-deepar-job-level-5')

我想以较小的学习率再次训练生成的模型。我遵循此处描述的增量训练方法:https://docs.aws.amazon.com/en_pv/sagemaker/latest/dg/incremental-training.html ,但它不起作用,显然(根据链接),只有两个内置模型支持增量学习。

有没有人找到解决方法,以便他们可以按照预定的学习率训练内置算法?

最佳答案

遗憾的是,SageMaker 内置 DeepAR 模型不支持学习率调度,也不支持增量学习。如果您想在 DeepAR 架构上实现学习率平台计划,我建议考虑:

关于amazon-web-services - 使用内置 sagemaker 算法进行增量学习,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/57946451/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com