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	docs: translated index page to korean (#20180)	2022-11-14 12:09:21 -05:00
_toctree.yml	[i18n-KO] Translated quicktour page to Korean (#20946)	2023-01-26 14:10:02 +01:00
in_translation.mdx	docs: translated index page to korean (#20180)	2022-11-14 12:09:21 -05:00
index.mdx	Resnet flax (#21472)	2023-03-24 19:45:57 +00:00
installation.mdx	🌐 [i18n-KO] Translated installation.mdx to Korean (#20948)	2023-01-18 10:05:23 +01:00
quicktour.mdx	[i18n-KO] Translated quicktour page to Korean (#20946)	2023-01-26 14:10:02 +01:00