📶
This commit is contained in:
parent
b097bdd620
commit
09e8972986
@ -1,9 +1,9 @@
|
|||||||
from tf_train import *
|
from tf_train import *
|
||||||
from sacred import Experiment
|
from sacred import Experiment
|
||||||
from sacred.observers import FileStorageObserver
|
from sacred.observers import MongoObserver
|
||||||
|
|
||||||
ex = Experiment('s452639')
|
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
|
@ex.config
|
||||||
def params():
|
def params():
|
||||||
|
@ -33,7 +33,7 @@ node {
|
|||||||
target: 'src/'
|
target: 'src/'
|
||||||
|
|
||||||
sh "cd src; python tf_train_sacred.py with epochs=${params.EPOCHS}"
|
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
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user