Update 'train.py'
This commit is contained in:
parent
e72ca8d99a
commit
3c469d8c83
2
train.py
2
train.py
@ -10,7 +10,7 @@ exint = Experiment('z-s434686-training')
|
||||
|
||||
exint.observers.append(MongoObserver(url='mongodb://admin:IUM_2021@172.17.0.1:27017',
|
||||
db_name='sacred'))
|
||||
exint.observers.append(FileStorageObserver('my_runs'))
|
||||
exint.observers.append(FileStorageObserver('z-s434686-training/master'))
|
||||
@exint.config
|
||||
def my_config():
|
||||
EPOCHS = int(os.environ['EPOCHS'])
|
||||
|
Loading…
Reference in New Issue
Block a user