fix bug for training

2024-09-12 15:11:09 +08:00
parent a79ca7749d
commit 4c69ed777b
15 changed files with 201 additions and 120 deletions
--- a/app_train.py
+++ b/app_train.py
@@ -1,8 +1,8 @@
 from PytorchBoot.application import PytorchBootApplication
-from runners.strategy_generator import StrategyGenerator
+from PytorchBoot.runners.trainer import DefaultTrainer

@PytorchBootApplication("train")
 class TrainApp:
    @staticmethod
    def start():
-        StrategyGenerator(r"configs\train_config.yaml").run()
+        DefaultTrainer("configs/train_config.yaml").run()