Changes to allow remote execution

2023-11-25 23:36:57 +00:00
parent 300f268286
commit a8db70e86d
7 changed files with 4500091 additions and 4 deletions
--- a/src/trainers/trainer.py
+++ b/src/trainers/trainer.py
@@ -76,6 +76,7 @@ class Trainer:
        task.connect(self.optimizer, name="optimizer")
        task.connect(self.criterion, name="criterion")
        task.connect(self.data_processor, name="data_processor")
+        task.connect(self, name="trainer")

        task.connect(self.data_processor.data_config, name="data_features")