如何在使用 keras 调谐器时使用我的自定义train_step？

问：

我的类中有一个自定义train_step和test_step函数。指定 keras 调谐器并运行 tuner.search 后，我收到一个错误，指定了不正确的大小，对我来说，这表明使用了默认的 fit 函数。

当我使用类的 model.fit 时，它确实有效，但对于调谐器，它不起作用。我唯一改变的是，我复制了我的类，并使其成为HyperModel而不是常规模型。

有没有办法让它在没有完全自定义训练循环的情况下使用自定义训练和测试函数？

编辑：另一个区别是您必须在构建函数中返回模型，这可能意味着它使用模型的拟合而不是类的拟合。