From b8340d0604f40a2bd907a88651f7a9932e815942 Mon Sep 17 00:00:00 2001 From: Alagris Date: Sun, 23 May 2021 20:14:05 +0200 Subject: [PATCH] fix --- train_model.py | 1 - 1 file changed, 1 deletion(-) diff --git a/train_model.py b/train_model.py index 6e8c283..fa57638 100644 --- a/train_model.py +++ b/train_model.py @@ -233,7 +233,6 @@ def run(kernel_size, hidden_layers, data_file, epochs, teacher_forcing_probabili train_model(cnn, learning_rate, in_alphabet, max_len, data, epochs, batch_size) torch.save(cnn.state_dict(), model_file) ex.add_artifact(model_file) - log_artifacts(model_file) mlflow.pytorch.log_model(cnn, "cnn-model", registered_model_name="PhoneticEdDistEmbeddings", signature=signature(cnn, in_alphabet, max_len))