transformers/tests/models/resnet
NielsRogge 0bae286de9
[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>
2022-11-28 17:20:24 +01:00
..
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_resnet.py [AutoBackbone] Improve API (#20407) 2022-11-28 17:20:24 +01:00
test_modeling_tf_resnet.py Add TF prefix to TF-Res test class (#18481) 2022-08-05 13:59:55 +02:00