fix
Some checks failed
s434749-training/pipeline/head There was a failure building this commit

This commit is contained in:
Alagris 2021-05-23 20:14:05 +02:00
parent aab43d29a5
commit b8340d0604

View File

@ -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) train_model(cnn, learning_rate, in_alphabet, max_len, data, epochs, batch_size)
torch.save(cnn.state_dict(), model_file) torch.save(cnn.state_dict(), model_file)
ex.add_artifact(model_file) ex.add_artifact(model_file)
log_artifacts(model_file)
mlflow.pytorch.log_model(cnn, "cnn-model", registered_model_name="PhoneticEdDistEmbeddings", mlflow.pytorch.log_model(cnn, "cnn-model", registered_model_name="PhoneticEdDistEmbeddings",
signature=signature(cnn, in_alphabet, max_len)) signature=signature(cnn, in_alphabet, max_len))