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
..
2023-03-20 17:07:31 +00:00
2022-05-11 23:35:07 -05:00
2023-01-14 09:25:05 +01:00
2022-08-06 09:38:55 +02:00
2022-12-12 09:23:23 -05:00
2022-05-31 08:41:54 -04:00
2023-03-08 07:54:30 -05:00
2023-01-14 09:25:05 +01:00
2022-08-08 15:53:43 -04:00
2022-10-12 13:18:11 -04:00
2022-11-07 13:57:39 -05:00
2022-12-12 10:38:56 -05:00
2022-05-11 22:25:44 -05:00
2023-03-24 19:45:57 +00:00
2022-05-31 08:41:54 -04:00
2023-02-06 09:26:22 -05:00
2022-04-04 10:25:46 -04:00
2022-05-11 20:47:50 -05:00
2022-11-09 15:34:08 +00:00
2023-02-07 12:27:31 -05:00
2022-10-12 08:28:54 -04:00
2023-02-23 09:41:28 +01:00
2022-10-11 08:14:21 -04:00
2022-05-25 19:10:16 -04:00
2023-03-14 12:08:14 +03:00
2022-12-29 02:12:40 -05:00