В Ray Tune, есть ли какое-нибудь руководство относительно того, можно ли использовать разные условия остановки для планировщика по сравнению с пробной версией?
Ниже я остановил асинхронный гиперполосный планировщик на основе neg_mean_loss
и настроил остановку на основе mean_f1
.
Должен ли я использовать одно и то же для обоих или это не имеет значения?
scheduler = schedulers.AsyncHyperBandScheduler(
time_attr='training_iteration',
reward_attr='neg_mean_loss', # <------
max_t=100,
grace_period=10,
reduction_factor=3,
brackets=3
)
all_trials = tune.run(
tune_trainable,
name="tuner",
scheduler=scheduler,
stop={"mean_f1": 0.99}, # <------
resources_per_trial={"cpu": 2, "gpu": 1},
config={"lr": tune.grid_search([0.0002, 0.003, 0.007, 0.01])
)