Add causal person fusion and MuVAP stage-3 baselines

2026-09-10 06:54:22 +08:00
parent 0534c9d3d0
commit d426b4740a
225 changed files with 27716 additions and 119 deletions
@@ -0,0 +1,11 @@
+__all__ = [
+    'build_criterion'
+]
+
+from utils import Registry
+
+CRITERIONS = Registry()
+
+def build_criterion(cfg, device=None):
+    criterion = CRITERIONS[cfg["loss"]](cfg)
+    return criterion.to(device)