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
..
tasks
_config.py
_toctree.yml	Added translation of serialization.mdx to Portuguese Issue #16824 (#19869)	2022-10-25 09:34:28 -04:00
accelerate.mdx
converting_tensorflow_models.mdx	Added translation of converting_tensorflow_models.mdx to Portuguese Issue #16824 (#19824)	2022-10-24 09:50:16 -04:00
create_a_model.mdx
custom_models.mdx	Replace awkward timm link with the expected one (#20109)	2022-11-07 13:57:39 -05:00
fast_tokenizers.mdx
index.mdx	Resnet flax (#21472)	2023-03-24 19:45:57 +00:00
installation.mdx
multilingual.mdx
pipeline_tutorial.mdx	Generate: move generation_*.py src files into generation/*.py (#20096)	2022-11-09 15:34:08 +00:00
quicktour.mdx
run_scripts.mdx	Added translation of run_scripts.mdx to Portuguese Issue #16824 (#19800)	2022-10-21 17:38:35 -04:00
serialization.mdx	Add ConvNeXT V2 (#21679)	2023-03-14 12:08:14 +03:00
training.mdx