mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-01 03:12:49 +08:00
further test fix
Summary: test_viewpool was inactive so missed being fixed in D36547815 (2d1c6d5d93
)
Reviewed By: kjchalup
Differential Revision: D36625587
fbshipit-source-id: e7224eadfa5581fe61f10f67d2221071783de04a
This commit is contained in:
parent
7c25d34d22
commit
f625fe1f8b
@ -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)
|
||||
|
Loading…
x
Reference in New Issue
Block a user