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