add data_splitor, modify dataset and encoder

2024-09-04 23:38:30 +08:00
parent e0fb9a7617
commit 129bcb872e
7 changed files with 148 additions and 50 deletions
--- a/runners/data_spliter.py
+++ b/runners/data_spliter.py
@@ -6,8 +6,8 @@ from PytorchBoot.utils import Log
 import PytorchBoot.stereotype as stereotype
 from PytorchBoot.status import status_manager

-@stereotype.runner("data_splitor")
-class DataSplitor(Runner):
+@stereotype.runner("data_spliter")
+class DataSpliter(Runner):
    def __init__(self, config):
        super().__init__(config)
        self.load_experiment("data_split")