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

..

_config.py

german docs translation (#18544)

2022-08-11 09:52:27 -04:00

_toctree.yml

german training, accelerate and model sharing (#19171)

2022-09-23 14:52:09 -04:00

accelerate.mdx

german training, accelerate and model sharing (#19171)

2022-09-23 14:52:09 -04:00

autoclass_tutorial.mdx

german autoclass (#19049)

2022-09-16 16:16:00 -04:00

index.mdx

Resnet flax (#21472)

2023-03-24 19:45:57 +00:00

installation.mdx

german docs translation (#18544)

2022-08-11 09:52:27 -04:00

model_sharing.mdx

Fix PushToHubCallback import in Share a model docs (#21457)

2023-02-06 09:26:22 -05:00

pipeline_tutorial.mdx

Generate: move generation_*.py src files into generation/*.py (#20096)

2022-11-09 15:34:08 +00:00

preprocessing.mdx

german processing (#19121)

2022-09-20 09:18:21 -04:00

quicktour.mdx

german docs translation (#18544)

2022-08-11 09:52:27 -04:00

training.mdx

Convert tokenizer outputs for Keras in doc example (#20732)

2022-12-12 16:14:04 +00:00