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

further test fix

Browse Source 
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:
Jeremy Reizenstein 2022-05-25 04:22:38 -07:00 committed by Facebook GitHub Bot
parent 7c25d34d22
commit f625fe1f8b
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/implicitron/test_forward_pass.py
View File

@ -163,6 +163,7 @@ class TestGenericModel(unittest.TestCase):
device = torch.device("cuda:1") device = torch.device("cuda:1")
args = get_default_args(GenericModel) args = get_default_args(GenericModel)
args.view_pooler_enabled = True args.view_pooler_enabled = True
args.image_feature_extractor_class_type = "ResNetFeatureExtractor"
args.image_feature_extractor_ResNetFeatureExtractor_args.add_masks = False args.image_feature_extractor_ResNetFeatureExtractor_args.add_masks = False
model = GenericModel(**args) model = GenericModel(**args)
model.to(device) model.to(device)