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