s487187/ium_487187
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

s

Browse Source
This commit is contained in:
Jakub Zaręba 2023-05-10 22:02:36 +02:00
parent 5e18b51413
commit a37c81cca6
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

@ -2,7 +2,7 @@ from sacred import Experiment
from sacred.observers import MongoObserver, FileStorageObserver
import os
os.environ["SACRED_NO_GIT"] = "1"
ex = Experiment('s487187-training', interactive=True, save_git_info=False)
ex.observers.append(MongoObserver(url='mongodb://admin:IUM_2021@172.17.0.1:27017', db_name='sacred'))
@ -10,7 +10,7 @@ ex.observers.append(MongoObserver(url='mongodb://admin:IUM_2021@172.17.0.1:27017
@ex.config
def my_config():
data_file = 'data.csv'
data_file = 'data.csv'
model_file = 'model.h5'
epochs = 10
batch_size = 32