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',
|
exint.observers.append(MongoObserver(url='mongodb://admin:IUM_2021@172.17.0.1:27017',
|
||||||
db_name='sacred'))
|
db_name='sacred'))
|
||||||
exint.observers.append(FileStorageObserver('my_runs'))
|
exint.observers.append(FileStorageObserver('z-s434686-training/master'))
|
||||||
@exint.config
|
@exint.config
|
||||||
def my_config():
|
def my_config():
|
||||||
EPOCHS = int(os.environ['EPOCHS'])
|
EPOCHS = int(os.environ['EPOCHS'])
|
||||||
|
Loading…
Reference in New Issue
Block a user