s426206/ium_426206
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Registred model.
Some checks failed
s426206-training/pipeline/head There was a failure building this commit
Details

Browse Source
This commit is contained in:
Jan Nowak 2021-05-23 15:52:15 +02:00
parent 8aa04f9fac
commit d12925a0d7
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Jenkinsfile_train
View File

@ -35,10 +35,10 @@ pipeline {
sh 'chmod +x dlgssdpytorch.py'
sh 'python3 ./dlgssdpytorch.py $PARAMETRY'
sh 'chmod +x train_mlflow.py'
sh 'python3 ./train_mlflow.py -e 5'
sh 'python3 ./train_mlflow.py $PARAMETRY'
//sh 'mlflow run --experiment-name s426206 .' //Uruchamiany projekt nie moze znajdowac sie w katalogach z wielkimi literami.
sh 'chmod +x generate_MLmodel.py'
sh 'python3 ./generate_MLmodel.py -e 5'
sh 'python3 ./generate_MLmodel.py $PARAMETRY'
}
}
}

3
generate_MLmodel.py
View File

@ -29,7 +29,8 @@ if __name__ == "__main__":
lr = args.lr
n_epochs = args.epochs
mlflow.set_tracking_uri("http://172.17.0.1:5000")
mlflow.set_tracking_uri("http://127.0.0.1:5000")
#mlflow.set_tracking_uri("http://172.17.0.1:5000")
mlflow.set_experiment("s426206")
with mlflow.start_run():
mlflow.log_param("lr", lr)