nbv_reconstruction/app_generate.py

10 lines
403 B
Python

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