Есть ли способ записать сводку модели keras в neptune?

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)

Они будут у вас на вкладке " Параметры " эксперимента, как в этом примере. Вы сможете добавить каждый параметр в виде столбца на панель управления для быстрого сравнения. Ищите зеленоватые столбцы на этой панели.