mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-02 03:42:50 +08:00
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
This commit is contained in:
parent
2d1c6d5d93
commit
d27ef14ec7
@ -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(
|
||||
|
Loading…
x
Reference in New Issue
Block a user