nbv_reconstruction/app_generate.py
2024-08-30 14:07:23 +08:00

10 lines
401 B
Python

from PytorchBoot.application import PytorchBootApplication
from runners.strategy_generator import StrategyGenerator
from runners.data_generator import DataGenerator
@PytorchBootApplication("generate")
class Generator:
@staticmethod
def start():
#StrategyGenerator("configs\strategy_generate_config.yaml").run()
DataGenerator("configs/data_generate_config.yaml").run()