add data_splitor, modify dataset and encoder

This commit is contained in:
hofee
2024-09-04 23:38:30 +08:00
parent e0fb9a7617
commit 129bcb872e
7 changed files with 148 additions and 50 deletions

View File

@@ -1,9 +1,9 @@
from PytorchBoot.application import PytorchBootApplication
from runners.data_splitor import DataSplitor
from runners.data_spliter import DataSpliter
@PytorchBootApplication("split")
class DataSplitApp:
@staticmethod
def start():
DataSplitor(r"configs\split_dataset_config.yaml").run()
DataSpliter(r"configs\split_dataset_config.yaml").run()