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