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

test_forward_pass: speedup and RE fix

Browse Source 
Summary: Use small image size for test_all_gm_configs

Reviewed By: shapovalov

Differential Revision: D36511528

fbshipit-source-id: 2c65f518a4f23626850343a62d103f85abfabd88
This commit is contained in:
Jeremy Reizenstein 2022-05-22 15:23:17 -07:00 committed by Facebook GitHub Bot
parent 2d1c6d5d93
commit d27ef14ec7
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/implicitron/test_forward_pass.py
View File

@ -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.implicitron.tools.config import expand_args_fields, get_default_args
from pytorch3d.renderer.cameras import look_at_view_transform, PerspectiveCameras 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): if os.environ.get("FB_TEST", False):
from common_testing import get_pytorch3d_dir from common_testing import get_pytorch3d_dir
@ -26,6 +30,10 @@ IMPLICITRON_CONFIGS_DIR = (
class TestGenericModel(unittest.TestCase): class TestGenericModel(unittest.TestCase):
@classmethod
def setUpClass(cls) -> None:
provide_resnet34()
def setUp(self): def setUp(self):
torch.manual_seed(42) torch.manual_seed(42)
@ -54,6 +62,8 @@ class TestGenericModel(unittest.TestCase):
for config_file in config_files: for config_file in config_files:
with self.subTest(name=config_file.stem): with self.subTest(name=config_file.stem):
cfg = _load_model_config_from_yaml(str(config_file)) cfg = _load_model_config_from_yaml(str(config_file))
cfg.render_image_height = 80
cfg.render_image_width = 80
model = GenericModel(**cfg) model = GenericModel(**cfg)
model.to(device) model.to(device)
self._one_model_test( self._one_model_test(