transformers
36b9a994 - Fix BeitFeatureExtractor postprocessing (#19119)
Commit
View On GitHub
Comment changes are shownComment changes are hidden
Commit
2 years ago
Fix BeitFeatureExtractor postprocessing (#19119) * return post-processed segmentations as list, add test * use torch to resize logits * fix assertion error if no target_size is specified
Author
alaradirik alaradirik
Parents
36e356ca
  • src/transformers/models/beit
    • File
      feature_extraction_beit.py
  • tests/models/beit
    • File
      test_modeling_beit.py
FAQ Terms Privacy Refunds Impressum