s487187/ium_487187
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

s

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
JenkinsfileDL
View File

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

4
train.py
View File

@ -1,7 +1,7 @@
from sacred import Experiment from sacred import Experiment
from sacred.observers import MongoObserver, FileStorageObserver 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.observers.append(MongoObserver(url='mongodb://admin:IUM_2021@172.17.0.1:27017', db_name='sacred'))
@ex.config @ex.config
@ -65,5 +65,5 @@ def run_experiment():
accuracy = train_model() accuracy = train_model()
ex.log_scalar('accuracy', accuracy) ex.log_scalar('accuracy', accuracy)
ex.add_artifact('model.h5') ex.add_artifact('model.h5')
ex.run() ex.run()