17 lines
594 B
Python
17 lines
594 B
Python
from PytorchBoot.application import PytorchBootApplication
|
|
from runners.inferencer import Inferencer
|
|
|
|
@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()
|