2

我目前正在使用 TFX 在带有 Kubeflow 引擎的 Google AI 平台上构建管道。我有一个模型,其中批量大小是需要调整的重要超参数。

我想在Tuner 组件中搜索这个超参数。

甚至可能吗?

我使用企鹅数据集的 TFX 示例,更准确地说是调谐器组件实现:在此处找到

_get_hyperparameters函数返回模型超参数的样本空间(见第 139 行)。但是,训练模型的批量大小是固定的,并在 tuner_fn 的末尾指定(参见第 246 行)。

有没有办法根据超参数空间中的样本动态更改批量大小?

4

0 回答 0