Update 'train.py'

2023-05-11 21:36:27 +02:00 · 2023-05-11 21:36:27 +02:00 · 65b73dc86f
commit 65b73dc86f
parent c6d3e5fa33
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -17,7 +17,7 @@ def my_config():
    

@exint.main
-def my_main(EPOCHS):
+def my_main(EPOCHS, _run):
    _run.info["epochs"] = EPOCHS
    normalize = layers.Normalization()
    train_data_x = exint.open_resource('./X_train.csv', 'r')