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
..
2022-06-06 07:47:34 -04:00
2022-05-13 07:55:44 -04:00
2022-10-25 09:34:28 -04:00
2022-08-06 09:38:55 +02:00
2022-10-24 09:50:16 -04:00
2022-07-26 08:01:08 -04:00
2022-11-07 13:57:39 -05:00
2022-05-13 16:18:14 -04:00
2023-03-24 19:45:57 +00:00
2022-05-13 07:55:44 -04:00
2022-05-13 07:55:44 -04:00
2022-11-09 15:34:08 +00:00
2022-06-07 17:35:05 -04:00
2022-10-21 17:38:35 -04:00
2023-03-14 12:08:14 +03:00
2022-06-17 20:06:05 -04:00