TensorFlow 2.x 中应使用 keras_tuner 而非手动 for 循环调用 model.fit() 由于其官方支持和自动管理,进行超参搜索 trial、内置 Hyperband 等高效算法,避免内存泄漏和不可复制的问题。TensorFlow 2.x 中该用
keras_tuner 而非手写循环
直接得出结论:不要自己写:不要自己写 for 循环调 model.fit() 做超参搜索,TensorFlow 官方推荐和维护的方案是 keras_tuner。它内置支持 BayesianOptimization、Hyperband、RandomSearch,能自动管理 trial 目录、检查点、早停和资源调度。
常见的错误是把握 Keras 将模型封装成函数后,使用 sklearn.model_selection.GridSearchCV 或自定义 for 循环跑-这将导致内存泄漏,GPU 不释放显存,不复制结果,不能使用 Hyperband 早停机制。
-
keras_tuner必须与tf.keras.Model(非Sequential或函数式 API 裸模型)一起使用,模型必须包装在一起build(hp)函数里 - 每个
trial默认新建一个独立的tf.keras.Model实例,避免权重污染 -
project_name和directory参数决定缓存路径,重复运行时自动 resume,但是,只有注意清目录,才能重新开始
Hyperband 比 BayesianOptimization 更适合深度学习任务
Hyperband 在有限的预算下更有效率:它首先快速训练大量简短 epoch 逐步淘汰表现不佳的模型,将资源集中在潜在配置上。而 BayesianOptimization 需要大量 trial 只有建模超参空间,才能对每个空间进行建模 trial 要求完整的训练,成本高,容易陷入局部最佳。
典型场景:你只有 8 小时 GPU 时间,想试 100 组超参——Hyperband 可能在 30–40 个 trial 内部收敛到良好的效果;BayesianOptimization 可能卡在前 20 个低效 trial 上。
立即学习“Python免费学习笔记(深入);
-
max_epochs为你完全训练 epoch 数(比如 100),factor默认控制“淘汰比例” 三、一般不需要改变 -
hyperband的objective必须是验证指标(如val_accuracy),不能是loss,否则,早停逻辑就会失效 - 若显存紧张,可在
tuner.search()里加workers=1和overwrite=True避免并发 OOM
hp.Int()、hp.Choice(),不能传 Python 原生类型
很多人把 learning_rate=0.001 写死在 compile() 里,或用 random.choice([1e-3, 1e-4]) ——这样 keras_tuner 这个参数根本看不见,不能优化。
Python 3.14.2
Python 3.14.2是Python编程语言于2025年12月5日发布的稳定版本,属于3.14系列的第二次维护更新。该版本包含18个修复项目,重点解决多过程、数据和正则表达模块的回归问题,修复CVE-2025-12084等安全漏洞。这个版本标志着Python发展的一个重要里程碑,即自由线程模式(删除GIL)正式得到官方支持。
下载所有待搜索参数必须通过 hp 对象声明,并在 build(hp) 函数内的动态值:
def build_model(hp):
model = tf.keras.Sequential([...])
lr = hp.Float('learning_rate', 1e-5, 1e-2, sampling='log')
model.compile(optimizer=tf.keras.optimizers.Adam(lr),
loss='sparse_categorical_crossentropy')
return model
-
hp.Float()用sampling='log'由于数量级变化比线性变化对学习率的影响更为合理 -
hp.Int('units', 32, 512, step=32)比hp.Choice('units', [32, 64, 128, 256])更灵活,但后者更容易控制搜索粒度 - 不要在
build()外定义hp变量,否则 tuner 依赖关系无法跟踪
tuner.get_best_models(1)[0]
tuner.get_best_models() 返回的是训练完成的模型对象,但在训练过程中没有保存 callbacks(比如 ModelCheckpoint),也不含 optimizer 状态-这意味着你不能继续训练或做 fine-tuning。
真正可复制和部署的最佳模型必须从 tuner.oracle.get_best_trials(1)[0].trial_id 对应的 checkpoint 目录加载:
best_trial = tuner.oracle.get_best_trials(1)[0]
best_model = tuner.hypermodel.build(best_trial.hyperparameters)
best_model.load_weights(f'{tuner.directory}/{tuner.project_name}/trial_{best_trial.trial_id}/checkpoints/epoch_XX')
- checkpoint 文件名里的
epoch_XX实际保存是最好的 epoch,不是最大 epoch 数 - 如果用了
EarlyStopping,记得在tuner.search()的callbacks否则,里显式传入 tuner 不知道该按哪一个 epoch 保存 -
tuner.get_best_hyperparameters(1)[0].values是可重复使用的超参字典,不要用打印日志肉眼抄袭
超参搜索真正困难的不是写几行代码,而是定义合理的搜索空间边界,选择正确的 early-stopping 的 patience、确认每一个 trial 无论随机种子是否被隔离——这些细节是否被遗漏,结果都是不可信的。