Commit f625fe1f authored by Jeremy Reizenstein's avatar Jeremy Reizenstein Committed by Facebook GitHub Bot
Browse files

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
parent 7c25d34d
......@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment