We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 61f3b9c commit 4768fe5Copy full SHA for 4768fe5
interact.py
@@ -74,12 +74,15 @@ def interact(newcfg: DictConfig) -> None:
74
logger.info(f"The topology will be {cfg.gender}.")
75
cfg.model.transforms.rots2joints.gender = cfg.gender
76
77
- logger.info("Loading data module")
78
- data_module = instantiate(cfg.data)
79
- logger.info(f"Data module '{cfg.data.dataname}' loaded")
+ if cfg.transforms.name == "XYZTransform":
+ nfeats = 64
+ elif cfg.transforms.name == "SMPLTransform":
80
+ nfeats = 135
81
+ else:
82
+ raise NotImplementedError()
83
84
model = instantiate(cfg.model,
- nfeats=data_module.nfeats,
85
+ nfeats=nfeats,
86
logger_name="none",
87
nvids_to_save=None,
88
_recursive_=False)
0 commit comments