From 100350d7e6ecd55ce9a53ac99b8d82f9bafd614c Mon Sep 17 00:00:00 2001 From: ydshieh Date: Wed, 31 Aug 2022 17:41:54 +0000 Subject: [PATCH] Add `feature_extractor_type` (#1) - Add `feature_extractor_type` (468ade778c7dfc810004d424c08f47a3b93c2001) Co-authored-by: Olivier Dehaene --- preprocessor_config.json | 1 + 1 file changed, 1 insertion(+) diff --git a/preprocessor_config.json b/preprocessor_config.json index 70fbc14..34a1af4 100644 --- a/preprocessor_config.json +++ b/preprocessor_config.json @@ -1,6 +1,7 @@ { "do_normalize": true, "do_resize": true, + "feature_extractor_type": "ViTFeatureExtractor", "image_mean": [ 0.5, 0.5,