assure that epochs is saved
Some checks failed
s444354-training/pipeline/head There was a failure building this commit
Some checks failed
s444354-training/pipeline/head There was a failure building this commit
This commit is contained in:
parent
718378d69a
commit
2a5e48f63c
@ -38,7 +38,7 @@ except:
|
|||||||
@ex.config
|
@ex.config
|
||||||
def my_config():
|
def my_config():
|
||||||
epochs = numberOfEpochParam
|
epochs = numberOfEpochParam
|
||||||
my_config()
|
|
||||||
|
|
||||||
|
|
||||||
dataframe_raw = pd.read_csv("winequality-red.csv")
|
dataframe_raw = pd.read_csv("winequality-red.csv")
|
||||||
@ -216,6 +216,7 @@ with open("result.txt", "w+") as file:
|
|||||||
@ex.main
|
@ex.main
|
||||||
def main():
|
def main():
|
||||||
lr = 1e-6
|
lr = 1e-6
|
||||||
|
my_config()
|
||||||
history5 = fit(epochs, lr, model, train_loader, val_loader)
|
history5 = fit(epochs, lr, model, train_loader, val_loader)
|
||||||
|
|
||||||
ex.run()
|
ex.run()
|
||||||
|
Loading…
Reference in New Issue
Block a user