Я имею в виду ссылки ниже, чтобы использовать Tensorboard в методе Sagemaker Script Mode.
https://www.tensorflow.org/tensorboard/get_started
Ниже приведен мой обратный вызов тензорной доски в моем учебном скрипте, который представляет собой файл .py
model = create_model()
model.compile(optimizer='adam',
loss='sparse_categorical_crossentropy',
metrics=['accuracy'])
log_dir = "logs/fit/" + datetime.datetime.now().strftime("%Y%m%d-%H%M%S")
tensorboard_callback = tf.keras.callbacks.TensorBoard(log_dir=log_dir, histogram_freq=1)
model.fit(x=x_train,
y=y_train,
epochs=5,
validation_data=(x_test, y_test),
callbacks=[tensorboard_callback])
В записной книжке я создаю приведенный ниже оценщик тензорного потока, в котором передаю указанное выше имя файла сценария как entry_point.
estimator = TensorFlow(
entry_point='Script_File.py',
train_instance_type=train_instance_type,
train_instance_count=1,
model_dir=model_dir,
hyperparameters=hyperparameters,
role=sagemaker.get_execution_role(),
base_job_name='tf-fashion-mnist',
framework_version='1.12.0',
py_version='py3',
output_path=<S3 Path>,
script_mode=True,
)
Я использую приведенный ниже код в своей записной книжке, чтобы начать обучение.
estimator.fit(inputs)
После завершения обучения я использую приведенный ниже код в Терминале (также пробовал в своей ячейке Блокнота) для запуска тензорной доски.
tensorboard --logdir logs/fit
Но на тензорной доске я не могу просматривать графики. Отображается сообщение Не удалось получить запуски. Что-то мне не хватает? Или мне нужно сделать какие-то дополнительные настройки в моем скрипте, чтобы увидеть мои журналы в Tensorboard?
