nbv_reconstruction/app_train.py

8 lines
277 B
Python
Raw Normal View History

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