Skip to content

Commit 4768fe5

Browse files
committedNov 8, 2022
remove dependancy of the data during the interact
1 parent 61f3b9c commit 4768fe5

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed
 

‎interact.py

+7-4
Original file line numberDiff line numberDiff line change
@@ -74,12 +74,15 @@ def interact(newcfg: DictConfig) -> None:
7474
logger.info(f"The topology will be {cfg.gender}.")
7575
cfg.model.transforms.rots2joints.gender = cfg.gender
7676

77-
logger.info("Loading data module")
78-
data_module = instantiate(cfg.data)
79-
logger.info(f"Data module '{cfg.data.dataname}' loaded")
77+
if cfg.transforms.name == "XYZTransform":
78+
nfeats = 64
79+
elif cfg.transforms.name == "SMPLTransform":
80+
nfeats = 135
81+
else:
82+
raise NotImplementedError()
8083

8184
model = instantiate(cfg.model,
82-
nfeats=data_module.nfeats,
85+
nfeats=nfeats,
8386
logger_name="none",
8487
nvids_to_save=None,
8588
_recursive_=False)

0 commit comments

Comments
 (0)