change DataLoadUtil and Dataset to blender version

This commit is contained in:
hofee
2024-08-29 13:54:13 -05:00
parent eceedd5c15
commit 80cd4aba9d
6 changed files with 135 additions and 129 deletions

8
app_train.py Normal file
View File

@@ -0,0 +1,8 @@
from PytorchBoot.application import PytorchBootApplication
from runners.strategy_generator import StrategyGenerator
@PytorchBootApplication("train")
class TrainApp:
@staticmethod
def start():
StrategyGenerator(r"configs\train_config.yaml").run()