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

parametrization of everything
Some checks failed
s444354-training/pipeline/head There was a failure building this commit
Details

Browse Source
This commit is contained in:
Adrian Charkiewicz 2022-05-08 15:11:30 +02:00
parent d3b41e9a96
commit 199b2f06fe
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
pytorch/pytorch.py
View File

@ -37,11 +37,6 @@ try:
except: except:
numberOfEpochParam = 1500 numberOfEpochParam = 1500
@ex.config
def my_config():
global epochs
epochs = numberOfEpochParam
@ -141,7 +136,14 @@ output_size = len(output_cols)
# In[10]: # In[10]:
@ex.config
def my_config():
global epochs
epochs = numberOfEpochParam
lr=lr
model=model
train_loader=train_loader
val_loader=val_loader
model=WineQuality() model=WineQuality()
@ -205,6 +207,6 @@ def main():
lr = 1e-6 lr = 1e-6
#my_config() #my_config()
#print("number of epochs is: ", epochs) #print("number of epochs is: ", epochs)
history5 = fit(lr, model, train_loader, val_loader) history5 = fit()
#ex.run() #ex.run()