我目前正在使用 TFX 在带有 Kubeflow 引擎的 Google AI 平台上构建管道。我有一个模型,其中批量大小是需要调整的重要超参数。
我想在Tuner 组件中搜索这个超参数。
甚至可能吗?
我使用企鹅数据集的 TFX 示例,更准确地说是调谐器组件实现:在此处找到。
该_get_hyperparameters
函数返回模型超参数的样本空间(见第 139 行)。但是,训练模型的批量大小是固定的,并在 tuner_fn 的末尾指定(参见第 246 行)。
有没有办法根据超参数空间中的样本动态更改批量大小?