From 15c0034ac334b444a2e08030d71a45db36335ec2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Patryk=20Ga=C5=82ka?= <patgal2@st.amu.edu.pl>
Date: Thu, 11 May 2023 21:32:10 +0200
Subject: [PATCH] Update 'train.py'

---
 train.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/train.py b/train.py
index 961924b..56ca6d8 100644
--- a/train.py
+++ b/train.py
@@ -15,11 +15,11 @@ exint.observers.append(FileStorageObserver('my_runs'))
 def my_config():
     EPOCHS = int(os.environ['EPOCHS'])
     train_data_x = exint.open_resource('./X_train.csv', 'r')
-    _run.info["epochs"] = EPOCHS
-
+    
 
 @exint.main
 def my_main(EPOCHS,train_data_x):
+    _run.info["epochs"] = EPOCHS
     normalize = layers.Normalization()
     games_all = train_data_x.copy()
     games_predict = train_data_x.pop('User_Score')