ray.tune.logger.aim.AimLoggerCallback.setup

AimLoggerCallback.setup(stop: Stopper | None = None, num_samples: int | None = None, total_num_samples: int | None = None, **info)

在训练开始时调用一次。

任何回调设置都应添加在此处（设置环境变量等）。

参数:

• stop – 停止标准。如果 time_budget_s 被传递给 train.RunConfig，一个 TimeoutStopper 将被传递到这里，无论是单独传递还是作为 CombinedStopper 的一部分。

• num_samples – 从超参数空间采样的次数。默认为1。如果提供了 grid_search 作为参数，网格将被重复 num_samples 次。如果此值为 -1，则（实际上）无限生成样本，直到满足停止条件。

• total_num_samples – 考虑网格搜索采样器的样本总数。

• **info – 用于向前兼容的 kwargs 字典。