423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-12-19 05:40:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

remove stray "generic_model_args" references

Browse Source 
Summary:
generic_model_args no longer exists. Update some references to it, mostly in doc.

This fixes the testing of all the yaml files in test_forward pass.

Reviewed By: shapovalov

Differential Revision: D38789202

fbshipit-source-id: f11417efe772d7f86368b3598aa66c52b1309dbf
This commit is contained in:
Jeremy Reizenstein
2022-08-18 07:18:55 -07:00
committed by Facebook GitHub Bot
parent d42e0d3d86
commit fdaaa299a7
4 changed files with 44 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
tests/implicitron/test_forward_pass.py
View File

@@ -207,10 +207,11 @@ def _load_model_config_from_yaml(config_path, strict=True) -> DictConfig:
def _load_model_config_from_yaml_rec(cfg: DictConfig, config_path: str) -> DictConfig:
cfg_loaded = OmegaConf.load(config_path)
if "generic_model_args" in cfg_loaded:
cfg_model_loaded = cfg_loaded.generic_model_args
else:
cfg_model_loaded = None
cfg_model_loaded = None
if "model_factory_ImplicitronModelFactory_args" in cfg_loaded:
factory_args = cfg_loaded.model_factory_ImplicitronModelFactory_args
if "model_GenericModel_args" in factory_args:
cfg_model_loaded = factory_args.model_GenericModel_args
defaults = cfg_loaded.pop("defaults", None)
if defaults is not None:
for default_name in defaults: