nbv_reconstruction/app_generate.py

10 lines
401 B
Python
Raw Normal View History

2024-08-21 17:11:56 +08:00
from PytorchBoot.application import PytorchBootApplication
from runners.strategy_generator import StrategyGenerator
2024-08-30 14:07:23 +08:00
from runners.data_generator import DataGenerator
2024-08-21 17:11:56 +08:00
@PytorchBootApplication("generate")
class Generator:
@staticmethod
def start():
2024-08-30 14:07:23 +08:00
#StrategyGenerator("configs\strategy_generate_config.yaml").run()
DataGenerator("configs/data_generate_config.yaml").run()