update pbnbv
This commit is contained in:
@@ -4,6 +4,8 @@ from runners.global_and_local_points_inferencer import GlobalAndLocalPointsInfer
|
||||
from runners.local_points_inferencer import LocalPointsInferencer
|
||||
from runners.inference_server import InferencerServer
|
||||
from runners.evaluate_uncertainty_guide import EvaluateUncertaintyGuide
|
||||
from runners.evaluate_pbnbv import EvaluatePBNBV
|
||||
|
||||
@PytorchBootApplication("global_points_inference")
|
||||
class GlobalPointsInferenceApp:
|
||||
@staticmethod
|
||||
@@ -46,6 +48,27 @@ class LocalPointsInferenceApp:
|
||||
'''
|
||||
LocalPointsInferencer("./configs/local/local_only_inference_config.yaml").run()
|
||||
|
||||
@PytorchBootApplication("real_global_only_inference")
|
||||
class RealGlobalOnlyInferenceApp:
|
||||
@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()
|
||||
'''
|
||||
GlobalPointsInferencer("./configs/local/real_global_only_inference_config.yaml").run()
|
||||
|
||||
@PytorchBootApplication("mlp_inference")
|
||||
class MLPInferenceApp:
|
||||
@staticmethod
|
||||
def start():
|
||||
GlobalAndLocalPointsInferencer("./configs/local/mlp_inference_config.yaml").run()
|
||||
|
||||
|
||||
@PytorchBootApplication("server")
|
||||
class InferenceServerApp:
|
||||
@staticmethod
|
||||
@@ -72,4 +95,10 @@ class EvaluateUncertaintyGuideApp:
|
||||
Trainer("path_to_your_train_config").run()
|
||||
Evaluator("path_to_your_eval_config").run()
|
||||
'''
|
||||
EvaluateUncertaintyGuide("./configs/local/uncertainty_guide_evaluation_config.yaml").run()
|
||||
EvaluateUncertaintyGuide("./configs/local/uncertainty_guide_evaluation_config.yaml").run()
|
||||
|
||||
@PytorchBootApplication("evaluate_pbnbv")
|
||||
class EvaluatePBNBVApp:
|
||||
@staticmethod
|
||||
def start():
|
||||
EvaluatePBNBV("./configs/local/pbnbv_evalutaion_config.yaml").run()
|
Reference in New Issue
Block a user