8 lines
269 B
Python
8 lines
269 B
Python
|
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()
|