eqweqwe

2021-05-15 20:28:43 +02:00 · 2021-05-15 20:28:43 +02:00 · aea643dee6
commit aea643dee6
parent 3180ec3ff9
1 changed files with 1 additions and 1 deletions
--- a/Zadanie_06_training.py
+++ b/Zadanie_06_training.py
@ -68,7 +68,7 @@ def prepare_model(epoch_param, batch_param, _run):
    NeuralModel.save('wine_model.h5')

@ex.main
-def my_main(train_size_param, test_size_param):
+def my_main(epoch_param, batch_param):
    print(prepare_model())