python-3.x - 有没有办法将 keras 模型摘要记录到海王星？

Question

我正在做一个 CNN 项目，我想将 model.summary 记录到 neptune.ai。这样做的目的是在比较不同模型时了解模型参数。任何帮助/提示将不胜感激！

Answer 1

您可以记录model.summary（假设它是 keras），如下所示：

neptune.init('workspace/project')
neptune.create_experiment()

model = keras.Sequential(...)
model.summary(print_fn=lambda x: neptune.log_text('model_summary', x))

这会将整个摘要记录为文本行。您可以稍后在实验的日志部分浏览它。在此示例中查找 tile:"model_summary" 。

另一种选择 - 为了更容易比较 - 是在实验创建时记录超参数，如下所示：

# Define parameters as Python dict
PARAMS = {'batch_size': 64,
          'n_epochs': 100,
          'shuffle': True,
          'activation': 'elu'}

# Pass PARAMS dict to params at experiment creation
neptune.create_experiment(params=PARAMS)

您将在实验的参数选项卡中拥有它们，就像在这个例子中一样。您将能够将每个参数作为列添加到仪表板以进行快速比较。在此仪表板中查找绿色列。