s444354/ium_444354
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

epochs no longer missing
All checks were successful
s444354-training/pipeline/head This commit looks good
Details
s444354-evaluation/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
Adrian Charkiewicz 2022-05-08 15:36:31 +02:00
parent 044740662e
commit e379116876
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pytorch/pytorch.py
View File

@ -138,8 +138,8 @@ output_size = len(output_cols)
# In[10]: # In[10]:
@ex.config @ex.config
def my_config(): def my_config():
global epochs #epochs = numberOfEpochParam
epochs = numberOfEpochParam epochs = 1500
lr=lr lr=lr
model=model model=model
train_loader=train_loader train_loader=train_loader