update pbnbv

This commit is contained in:
2025-05-19 16:32:04 +08:00
parent b98753bfbb
commit cd436605a5
25 changed files with 1228 additions and 178 deletions

View File

@@ -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()