simplify image_feature_extractor control

Summary: If no view pooling, don't disable image_feature_extractor. Make image_feature_extractor default to absent.

Reviewed By: davnov134

Differential Revision: D36547815

fbshipit-source-id: e51718e1bcbf65b8b365a6e894d4324f136635e9
This commit is contained in:
Jeremy Reizenstein
2022-05-20 08:32:19 -07:00
committed by Facebook GitHub Bot
parent 9fe15da3cd
commit 2d1c6d5d93
2 changed files with 2 additions and 13 deletions

View File

@@ -69,6 +69,7 @@ class TestGenericModel(unittest.TestCase):
args.view_pooler_args.feature_aggregator_class_type = (
"AngleWeightedIdentityFeatureAggregator"
)
args.image_feature_extractor_class_type = "ResNetFeatureExtractor"
args.implicit_function_class_type = "IdrFeatureField"
idr_args = args.implicit_function_IdrFeatureField_args
idr_args.n_harmonic_functions_xyz = 1729