release code

Dassl.ProGrad.pytorch/dassl/evaluation/build.py

@@ -0,0 +1,11 @@
+from dassl.utils import Registry, check_availability
+
+EVALUATOR_REGISTRY = Registry("EVALUATOR")
+
+
+def build_evaluator(cfg, **kwargs):
+    avai_evaluators = EVALUATOR_REGISTRY.registered_names()
+    check_availability(cfg.TEST.EVALUATOR, avai_evaluators)
+    if cfg.VERBOSE:
+        print("Loading evaluator: {}".format(cfg.TEST.EVALUATOR))
+    return EVALUATOR_REGISTRY.get(cfg.TEST.EVALUATOR)(cfg, **kwargs)