Flax Regnet (#21867)

* initial commit * review changes * post model PR merge * updating doc
2023-04-04 22:11:12 +05:30
parent fc5b7419d4
commit 900677487d
17 changed files with 1136 additions and 12 deletions
--- a/docs/source/en/model_doc/regnet.mdx
+++ b/docs/source/en/model_doc/regnet.mdx
@@ -67,4 +67,16 @@ If you're interested in submitting a resource to be included here, please feel f
 ## TFRegNetForImageClassification

 [[autodoc]] TFRegNetForImageClassification
-    - call
+    - call
+
+
+## FlaxRegNetModel
+
+[[autodoc]] FlaxRegNetModel
+    - __call__
+
+
+## FlaxRegNetForImageClassification
+
+[[autodoc]] FlaxRegNetForImageClassification
+    - __call__