From f625fe1f8bfbbf393569f2e768087e4cca136c66 Mon Sep 17 00:00:00 2001 From: Jeremy Reizenstein Date: Wed, 25 May 2022 04:22:38 -0700 Subject: [PATCH] further test fix Summary: test_viewpool was inactive so missed being fixed in D36547815 (https://github.com/facebookresearch/pytorch3d/commit/2d1c6d5d9382651bbf825f5f1677d576305d1f92) Reviewed By: kjchalup Differential Revision: D36625587 fbshipit-source-id: e7224eadfa5581fe61f10f67d2221071783de04a --- tests/implicitron/test_forward_pass.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/implicitron/test_forward_pass.py b/tests/implicitron/test_forward_pass.py index 8f456dd9..52d5016e 100644 --- a/tests/implicitron/test_forward_pass.py +++ b/tests/implicitron/test_forward_pass.py @@ -163,6 +163,7 @@ class TestGenericModel(unittest.TestCase): device = torch.device("cuda:1") args = get_default_args(GenericModel) args.view_pooler_enabled = True + args.image_feature_extractor_class_type = "ResNetFeatureExtractor" args.image_feature_extractor_ResNetFeatureExtractor_args.add_masks = False model = GenericModel(**args) model.to(device)