fix

2021-05-23 20:14:05 +02:00 · 2021-05-23 20:14:05 +02:00 · b8340d0604
commit b8340d0604
parent aab43d29a5
1 changed files with 0 additions and 1 deletions
--- 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))