s
This commit is contained in:
parent
4297ef2600
commit
6379116689
@ -27,7 +27,7 @@ pipeline {
|
||||
stage('Trenuj model') {
|
||||
steps {
|
||||
script {
|
||||
sh "python3 train.py" //--epochs $EPOCHS
|
||||
sh "python3 train.py --epochs $EPOCHS"
|
||||
}
|
||||
}
|
||||
}
|
||||
|
4
train.py
4
train.py
@ -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()
|
Loading…
Reference in New Issue
Block a user