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

Added lr again
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:31:47 +02:00
parent 709971cc1c
commit 044740662e
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pytorch/pytorch.py
View File

@ -87,7 +87,7 @@ train_ds, val_ds = random_split(dataset, [1300, 299])
batch_size=50 batch_size=50
train_loader = DataLoader(train_ds, batch_size, shuffle=True) train_loader = DataLoader(train_ds, batch_size, shuffle=True)
val_loader = DataLoader(val_ds, batch_size) val_loader = DataLoader(val_ds, batch_size)
lr = 1e-6
# In[8]: # In[8]:
@ -140,6 +140,7 @@ output_size = len(output_cols)
def my_config(): def my_config():
global epochs global epochs
epochs = numberOfEpochParam epochs = numberOfEpochParam
lr=lr
model=model model=model
train_loader=train_loader train_loader=train_loader
val_loader=val_loader val_loader=val_loader
@ -203,9 +204,8 @@ with open("result.txt", "w+") as file:
@ex.automain @ex.automain
def main(): def main():
lr = 1e-6
#my_config() #my_config()
#print("number of epochs is: ", epochs) #print("number of epochs is: ", epochs)
history5 = fit(lr) history5 = fit()
#ex.run() #ex.run()