transformers
0bae286d - [AutoBackbone] Improve API (#20407)

Commit
3 years ago
[AutoBackbone] Improve API (#20407) * Add hidden states and attentions to backbone outputs * Update ResNet * Fix more tests * Debug test * Fix test_determinism * Fix test_save_load * Remove file * Disable fx tests * Test * Add fx support for backbones Co-authored-by: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
Author
Parents
Loading