Update 'train.py'
This commit is contained in:
parent
c6d3e5fa33
commit
65b73dc86f
2
train.py
2
train.py
@ -17,7 +17,7 @@ def my_config():
|
|||||||
|
|
||||||
|
|
||||||
@exint.main
|
@exint.main
|
||||||
def my_main(EPOCHS):
|
def my_main(EPOCHS, _run):
|
||||||
_run.info["epochs"] = EPOCHS
|
_run.info["epochs"] = EPOCHS
|
||||||
normalize = layers.Normalization()
|
normalize = layers.Normalization()
|
||||||
train_data_x = exint.open_resource('./X_train.csv', 'r')
|
train_data_x = exint.open_resource('./X_train.csv', 'r')
|
||||||
|
Loading…
Reference in New Issue
Block a user