docs(akida_models_apis): add transformers and AkidaUNet in zoo

Brainchip-Inc · Nov 15, 2022 · d481443 · d481443
1 parent 06c24ab
commit d481443
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/doc/api_reference/akida_models_apis.rst b/doc/api_reference/akida_models_apis.rst
@@ -112,6 +112,13 @@ Akida models API
     .. autofunction:: akida_models.imagenet.preprocessing.preprocess_image
     .. autofunction:: akida_models.imagenet.preprocessing.index_to_label
 
+    AkidaUNet
+    ---------
+
+    Portait128
+    ~~~~~~~~~~
+    .. autofunction:: akida_models.akida_unet_portrait128
+
     Mobilenet
     ---------
 
@@ -212,3 +219,16 @@ Akida models API
     ~~~~~
     .. autofunction:: akida_models.gxnor_mnist
     .. autofunction:: akida_models.gxnor_mnist_pretrained
+
+    Transformers
+    ------------
+
+    ImageNet
+    ~~~~~~~~
+    .. autofunction:: akida_models.deit_imagenet
+    .. autofunction:: akida_models.bc_deit_ti16
+    .. autofunction:: akida_models.bc_deit_ti16_imagenet_pretrained
+    .. autofunction:: akida_models.bc_deit_dist_ti16_imagenet_pretrained
+    .. autofunction:: akida_models.vit_imagenet
+    .. autofunction:: akida_models.bc_vit_ti16
+    .. autofunction:: akida_models.bc_vit_ti16_imagenet_pretrained