Update 'train.py'

This commit is contained in:
Patryk Gałka 2023-05-11 22:12:38 +02:00
parent e72ca8d99a
commit 3c469d8c83

View File

@ -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'])