fix batch bug

Author: LukyanovKirillML

src/models_builder/gnn_models.py

@@ -808,7 +808,7 @@ def train_1_step(self, gen_dataset):
         loss = 0
         for batch in loader:
             self.before_batch(batch)
-            loss += self.train_on_batch(batch, task_type)
+            loss += self.train_on_batch_full(batch, task_type)
             self.after_batch(batch)
         print("loss %.8f" % loss)
         self.modification.epochs += 1