Add causal person fusion and MuVAP stage-3 baselines
This commit is contained in:
+9
@@ -0,0 +1,9 @@
|
||||
__all__ = ['build_model']
|
||||
|
||||
from utils import Registry
|
||||
|
||||
META_ARCHITECTURES = Registry()
|
||||
|
||||
def build_model(cfg, device=None):
|
||||
model = META_ARCHITECTURES[cfg["model"]](cfg)
|
||||
return model.to(device)
|
||||
Reference in New Issue
Block a user