2024-08-29 13:54:13 -05:00
|
|
|
from PytorchBoot.application import PytorchBootApplication
|
2024-09-12 15:11:09 +08:00
|
|
|
from PytorchBoot.runners.trainer import DefaultTrainer
|
2024-08-29 13:54:13 -05:00
|
|
|
|
|
|
|
@PytorchBootApplication("train")
|
|
|
|
class TrainApp:
|
|
|
|
@staticmethod
|
|
|
|
def start():
|
2024-09-25 09:31:22 +00:00
|
|
|
DefaultTrainer("configs/server/server_train_config.yaml").run()
|