Ray学习之scheduler的使用

最新推荐文章于 2025-03-07 00:24:19 发布

构建的乐趣

最新推荐文章于 2025-03-07 00:24:19 发布

阅读量363

点赞数 1

文章标签： ray

本文链接：https://blog.csdn.net/qq_44065334/article/details/123390018

版权

Ray包含 data train tune三个模块，还是非常全面的。

    m_scheduler = tune.schedulers.MedianStoppingRule(time_attr='training_iteration',
     metric = "stress_err", mode = "min", 
     grace_period = 10, min_samples_required = 5, 
     min_time_slice = 0, hard_stop = True)

    result = tune.run(
        trainOneParameter,
        config=config,
        metric="stress_err",
        mode="min",
        num_samples=num_samples,
        resources_per_trial={'gpu': 1},
        scheduler=m_scheduler
    )

如上代码段，报错：
ValueError: You passed a metric or mode argument to tune.run(), but the scheduler you are using was already instantiated with their own metric and mode parameters. Either remove the arguments from your scheduler or from your call to tune.run()
只需删去run中的mode,metric即可，scheduler已经帮你做好mode和metric的工作了。