有没有办法将 keras 模型摘要记录到海王星？

Question

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

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))

这会将整个摘要记录为文本行。您稍后可以在实验的 Logs 部分浏览它。在此 example.

中查找图块：“model_summary”

为了便于比较，另一种选择是在实验创建时记录 hyper-parameters，如下所示：

# 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)

您将在实验的参数选项卡中找到它们，就像这个 example. You will be able to add each parameter as a column to the dashboard for quick compare. Look for greenish columns in this dashboard.

有没有办法将 keras 模型摘要记录到海王星？

Is there a way to log the keras model summary to neptune?

python-3.x

deep-learning

keras

neptune