This commit is contained in:
Jakub Zaręba 2023-05-10 21:39:27 +02:00
parent 4297ef2600
commit 6379116689
2 changed files with 3 additions and 3 deletions

View File

@ -27,7 +27,7 @@ pipeline {
stage('Trenuj model') {
steps {
script {
sh "python3 train.py" //--epochs $EPOCHS
sh "python3 train.py --epochs $EPOCHS"
}
}
}

View File

@ -1,7 +1,7 @@
from sacred import Experiment
from sacred.observers import MongoObserver, FileStorageObserver
ex = Experiment('s487187-training', interactive=True)
ex = Experiment('s449312-training_master', interactive=True)
ex.observers.append(MongoObserver(url='mongodb://admin:IUM_2021@172.17.0.1:27017', db_name='sacred'))
@ex.config
@ -65,5 +65,5 @@ def run_experiment():
accuracy = train_model()
ex.log_scalar('accuracy', accuracy)
ex.add_artifact('model.h5')
ex.run()