423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-12-21 14:50:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix test_config_use

Browse Source 
Summary: Fixes to reenable test_create_gm_overrides. Followup from D35852367 (47d06c8924) using logic from D36349361 (9e57b994ca).

Reviewed By: shapovalov

Differential Revision: D36371762

fbshipit-source-id: ad5fbbb4b5729fac41980d118f17a2589f7e6aba
This commit is contained in:
Jeremy Reizenstein
2022-05-13 07:15:26 -07:00
committed by Facebook GitHub Bot
parent 2c1901522a
commit b5f3d3ce12
2 changed files with 11 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
tests/implicitron/test_config_use.py
View File

@@ -32,9 +32,9 @@ from pytorch3d.implicitron.tools.config import (
if os.environ.get("FB_TEST", False):
from common_testing import get_tests_dir
from .common_resources import provide_lpips_vgg
from .common_resources import provide_resnet34
else:
from common_resources import provide_lpips_vgg # noqa
from common_resources import provide_resnet34
from tests.common_testing import get_tests_dir
DATA_DIR = get_tests_dir() / "implicitron/data"
@@ -48,7 +48,6 @@ class TestGenericModel(unittest.TestCase):
self.maxDiff = None
def test_create_gm(self):
provide_lpips_vgg()
args = get_default_args(GenericModel)
gm = GenericModel(**args)
self.assertIsInstance(gm.renderer, MultiPassEmissionAbsorptionRenderer)
@@ -60,8 +59,8 @@ class TestGenericModel(unittest.TestCase):
self.assertIsNone(gm.view_pooler)
self.assertIsNone(gm.image_feature_extractor)
def _test_create_gm_overrides(self):
provide_lpips_vgg()
def test_create_gm_overrides(self):
provide_resnet34()
args = get_default_args(GenericModel)
args.view_pooler_enabled = True
args.image_feature_extractor_enabled = True