我正在使用 Watson Studio 的 HPO 功能训练一个简单的 MLP。但是,在查看我的日志时,指标没有显示。运行非 HPO 训练运行时,指标日志记录有效,但在 HPO 中运行时日志不显示。
这是我定义 Tensorboard 回调的方式:
tb_directory = os.path.join(os.environ["JOB_STATE_DIR"], "logs", "tb",
os.makedirs(tb_directory, exist_ok=True)
tensorboard = TensorBoard(log_dir=tb_directory)
history = model.fit(x_train, y_train,
batch_size=batch_size,
epochs=epochs,
validation_data=(x_test, y_test),
callbacks=[tensorboard])