8 lines
263 B
Python
8 lines
263 B
Python
from PytorchBoot.application import PytorchBootApplication
|
|
from PytorchBoot.runners.trainer import DefaultTrainer
|
|
|
|
@PytorchBootApplication("train")
|
|
class TrainApp:
|
|
@staticmethod
|
|
def start():
|
|
DefaultTrainer("configs/train_config.yaml").run() |