add split dataset

2024-09-02 18:21:38 +08:00
parent f58360c0c0
commit 2fcfcd1966
5 changed files with 111 additions and 9 deletions
--- a/app_split.py
+++ b/app_split.py
@@ -0,0 +1,9 @@
+from PytorchBoot.application import PytorchBootApplication
+from runners.data_splitor import DataSplitor
+
+@PytorchBootApplication("split")
+class DataSplitApp:
+    @staticmethod
+    def start():
+        DataSplitor(r"configs\split_dataset_config.yaml").run()
+