From 09fd3c44ab5a286ec57887c33636db602cc5b3dd Mon Sep 17 00:00:00 2001 From: Alagris Date: Sun, 23 May 2021 19:46:14 +0200 Subject: [PATCH] fix --- train-model.Jenkinsfile | 2 +- train_model.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/train-model.Jenkinsfile b/train-model.Jenkinsfile index b51c215..d3ccc0d 100644 --- a/train-model.Jenkinsfile +++ b/train-model.Jenkinsfile @@ -10,7 +10,7 @@ pipeline { stages { stage('Build') { steps { - cleanWs() +// cleanWs() git 'https://git.wmi.amu.edu.pl/s434749/ium_434749.git' copyArtifacts fingerprintArtifacts: true, projectName: 's434749-create-dataset', selector: lastSuccessful() sh "python3 train_model.py with 'batch_size=${params.batch_size}' 'learning_rate=${params.learning_rate}' 'epochs=${params.epochs}'" diff --git a/train_model.py b/train_model.py index 76ac7ed..8fe1dff 100644 --- a/train_model.py +++ b/train_model.py @@ -193,10 +193,10 @@ def cfg(): def signature(model,in_alphabet,max_len): - mock_x = [('abc', 'xyz'), ('hey', 'man')] + mock_x = [('abc', 'xyz')] mock_text, _ = encode_str(mock_x, in_alphabet, max_len) mock_y = model(mock_text) - return mlflow.models.signature.infer_signature(mock_x, mock_y) + return mlflow.models.signature.infer_signature(mock_text, mock_y) @ex.automain def run(kernel_size, hidden_layers, data_file, epochs, teacher_forcing_probability, learning_rate, batch_size, max_len,