change DataLoadUtil and Dataset to blender version
This commit is contained in:
8
app_train.py
Normal file
8
app_train.py
Normal 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()
|
Reference in New Issue
Block a user