nbv_reconstruction/app_inference.py

31 lines
1.1 KiB
Python
Raw Normal View History

2024-09-19 00:14:26 +08:00
from PytorchBoot.application import PytorchBootApplication
from runners.inferencer import Inferencer
2024-11-02 21:54:46 +00:00
from runners.inference_server import InferencerServer
2024-09-19 00:14:26 +08:00
@PytorchBootApplication("inference")
class InferenceApp:
@staticmethod
def start():
'''
call default or your custom runners here, code will be executed
automatically when type "pytorch-boot run" or "ptb run" in terminal
example:
Trainer("path_to_your_train_config").run()
Evaluator("path_to_your_eval_config").run()
'''
Inferencer("./configs/local/inference_config.yaml").run()
2024-11-02 21:54:46 +00:00
@PytorchBootApplication("server")
class InferenceServerApp:
@staticmethod
def start():
'''
call default or your custom runners here, code will be executed
automatically when type "pytorch-boot run" or "ptb run" in terminal
example:
Trainer("path_to_your_train_config").run()
Evaluator("path_to_your_eval_config").run()
'''
InferencerServer("./configs/server/server_inference_server_config.yaml").run()