diff --git a/learning/Jenkinsfile b/learning/Jenkinsfile index 43b5ef5..ebdc731 100644 --- a/learning/Jenkinsfile +++ b/learning/Jenkinsfile @@ -23,6 +23,7 @@ pipeline { sh 'python ./learning/ml.py' sh "export MLFLOW_TRACKING_URI=http://172.17.0.1:5000" sh "python ./learning/ml-mlflow.py" + sh "ls -l" } } stage('archiveArtifacts') { diff --git a/learning/ml-mlflow.py b/learning/ml-mlflow.py index 291dcbf..d4bdb1b 100644 --- a/learning/ml-mlflow.py +++ b/learning/ml-mlflow.py @@ -80,10 +80,10 @@ with mlflow.start_run(): mlflow.set_experiment("s434700") tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme if tracking_url_type_store != "file": - mlflow.pytorch.log_model(model, "model.pt", registered_model_name="s434700", signature=signature, + mlflow.pytorch.log_model(model, "cs_model", registered_model_name="s434700", signature=signature, input_example=test_input) else: - mlflow.pytorch.log_model(model, "model.pt", + mlflow.pytorch.log_model(model, "cs_model", signature=signature, input_example=test_input) mlflow.pytorch.save_model( - model, "model.pt", signature=signature, input_example=test_input) + model, "cs_model", signature=signature, input_example=test_input)