transformers/docs/source
Shubhamai a0cbbba31f
Resnet flax (#21472)
* [WIP] flax resnet

* added pretrained flax models, results reproducible

* Added pretrained flax models, results reproducible

* working on tests

* no real code change, just some comments

* [flax] adding support for batch norm layers

* fixing bugs related to pt+flax integration

* removing loss from modeling flax output class

* fixing classifier tests

* fixing comments, model output

* cleaning comments

* review changes

* review changes

* Apply suggestions from code review

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* renaming Flax to PyTorch

---------

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
2023-03-24 19:45:57 +00:00
..
de Resnet flax (#21472) 2023-03-24 19:45:57 +00:00
en Resnet flax (#21472) 2023-03-24 19:45:57 +00:00
es Resnet flax (#21472) 2023-03-24 19:45:57 +00:00
fr Add ConvNeXT V2 (#21679) 2023-03-14 12:08:14 +03:00
it Resnet flax (#21472) 2023-03-24 19:45:57 +00:00
ja Resnet flax (#21472) 2023-03-24 19:45:57 +00:00
ko Resnet flax (#21472) 2023-03-24 19:45:57 +00:00
pt Resnet flax (#21472) 2023-03-24 19:45:57 +00:00
zh Add ConvNeXT V2 (#21679) 2023-03-14 12:08:14 +03:00
_config.py Adding evaluate to the list of libraries required in generated notebooks (#20850) 2022-12-21 14:04:08 +01:00