This commit is contained in:
Robert Bendun 2023-05-15 04:00:28 +02:00
parent b097bdd620
commit 09e8972986
2 changed files with 3 additions and 3 deletions

View File

@ -1,9 +1,9 @@
from tf_train import *
from sacred import Experiment
from sacred.observers import FileStorageObserver
from sacred.observers import MongoObserver
ex = Experiment('s452639')
ex.observers.append(FileStorageObserver('experiments'))
ex.observers.append(MongoObserver(url="mongodb://admin:IUM_2021@172.17.0.1:27017", db_name='sacred'))
@ex.config
def params():

View File

@ -33,7 +33,7 @@ node {
target: 'src/'
sh "cd src; python tf_train_sacred.py with epochs=${params.EPOCHS}"
archiveArtifacts artifacts: 'src/model.keras,src/experiments/**/*', followSymlinks: false
archiveArtifacts artifacts: 'src/model.keras', followSymlinks: false
}
}