s434749/ium_434749
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

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

Browse Source
This commit is contained in:
Alagris 2021-05-23 20:14:05 +02:00
parent aab43d29a5
commit b8340d0604
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
train_model.py
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))