release code

2026-10-12 22:35:10 +08:00
commit c1e3081dfd
291 changed files with 60190 additions and 0 deletions
@@ -0,0 +1,11 @@
+from dassl.utils import Registry, check_availability
+
+TRAINER_REGISTRY = Registry("TRAINER")
+
+
+def build_trainer(cfg):
+    avai_trainers = TRAINER_REGISTRY.registered_names()
+    check_availability(cfg.TRAINER.NAME, avai_trainers)
+    if cfg.VERBOSE:
+        print("Loading trainer: {}".format(cfg.TRAINER.NAME))
+    return TRAINER_REGISTRY.get(cfg.TRAINER.NAME)(cfg)