From d27ef14ec7a42a0129d5ca4d94b6fe4903876c9f Mon Sep 17 00:00:00 2001 From: Jeremy Reizenstein Date: Sun, 22 May 2022 15:23:17 -0700 Subject: [PATCH] test_forward_pass: speedup and RE fix Summary: Use small image size for test_all_gm_configs Reviewed By: shapovalov Differential Revision: D36511528 fbshipit-source-id: 2c65f518a4f23626850343a62d103f85abfabd88 --- tests/implicitron/test_forward_pass.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/tests/implicitron/test_forward_pass.py b/tests/implicitron/test_forward_pass.py index 6755f04a..8f456dd9 100644 --- a/tests/implicitron/test_forward_pass.py +++ b/tests/implicitron/test_forward_pass.py @@ -14,6 +14,10 @@ from pytorch3d.implicitron.models.renderer.base import EvaluationMode from pytorch3d.implicitron.tools.config import expand_args_fields, get_default_args from pytorch3d.renderer.cameras import look_at_view_transform, PerspectiveCameras +if os.environ.get("FB_TEST", False): + from .common_resources import provide_resnet34 +else: + from common_resources import provide_resnet34 if os.environ.get("FB_TEST", False): from common_testing import get_pytorch3d_dir @@ -26,6 +30,10 @@ IMPLICITRON_CONFIGS_DIR = ( class TestGenericModel(unittest.TestCase): + @classmethod + def setUpClass(cls) -> None: + provide_resnet34() + def setUp(self): torch.manual_seed(42) @@ -54,6 +62,8 @@ class TestGenericModel(unittest.TestCase): for config_file in config_files: with self.subTest(name=config_file.stem): cfg = _load_model_config_from_yaml(str(config_file)) + cfg.render_image_height = 80 + cfg.render_image_width = 80 model = GenericModel(**cfg) model.to(device) self._one_model_test(