Mlflow не регистрирует артефакты автоматически при регистрации изображений ⇐ Python

[Anonymous]

Для своих экспериментов я использую бесплатный сервер MLFlow Dagshub. Я столкнулся с каким-то странным поведением. Когда я запускаю простую модель Keras, подходящую для MLFlow Autolog, вот так:

Код: Выделить всё

mlflow.tensorflow.autolog()
dagshub.init('my_experiment_name',
'my_user_name',
mlflow=True)

# Define the parameters.
num_epochs = 10
batch_size = 256

# Train the model.
history = model.fit(X_train,
y_train,
epochs=num_epochs,
batch_size=batch_size,
validation_data=(X_test, y_test))
mlflow.end_run()

Это приводит к ожидаемому поведению. Я вижу модель в артефактах и ​​показателях.


Однако, когда я пытаюсь добавить цифру, содержащую точность обучения и потери, артефакты содержат только изображения.

Код: Выделить всё

mlflow.tensorflow.autolog()
dagshub.init('my_experiment_name',
'my_user_name',
mlflow=True)

# Define the parameters.
num_epochs = 10
batch_size = 256

# Train the model.
history = model.fit(X_train,
y_train,
epochs=num_epochs,
batch_size=batch_size,
validation_data=(X_test, y_test))

##_________ Problematic Part
fig, ax = plt.subplots(1,2,figsize=(10,4))
ax[0].plot(history.history['accuracy'], label='Accuracy' )
ax[0].plot(history.history['val_accuracy'], label='Val Accuracy' )
ax[0].set_title('Accuracy')
ax[0].legend(loc='best')
ax[1].plot(history.history['loss'], label='Loss' )
ax[1].plot(history.history['val_loss'], label='Val Loss' )
ax[1].set_title('Loss')
ax[1].legend(loc='best')
mlflow.log_figure(fig,'training_history.png')
# _________

mlflow.end_run()

Артефакты модели отсутствуют. Метрики также не записываются.
Я что-то упустил?

Пожалуйста, помогите.

Подробнее здесь: https://stackoverflow.com/questions/786 ... ing-images