From 06c24abf520d9179b55196bc5c400511c2f95f87 Mon Sep 17 00:00:00 2001 From: ktsiknos Date: Tue, 8 Nov 2022 18:07:29 +0100 Subject: [PATCH] docs(akida_models_apis): add new components Note that Distiller members were removed as the inherited members use the native keras documentation that will display badly. --- doc/api_reference/akida_models_apis.rst | 36 ++++++++++++++++++++++++- 1 file changed, 35 insertions(+), 1 deletion(-) diff --git a/doc/api_reference/akida_models_apis.rst b/doc/api_reference/akida_models_apis.rst index a35cc7ed8..d76270324 100644 --- a/doc/api_reference/akida_models_apis.rst +++ b/doc/api_reference/akida_models_apis.rst @@ -20,6 +20,31 @@ Akida models API ----------- .. autofunction:: akida_models.layer_blocks.dense_block + mlp_block + --------- + .. autofunction:: akida_models.layer_blocks.mlp_block + + multi_head_attention + -------------------- + .. autofunction:: akida_models.layer_blocks.multi_head_attention + + transformer_block + ----------------- + .. autofunction:: akida_models.layer_blocks.transformer_block + + conv_transpose_block + -------------------- + .. autofunction:: akida_models.layer_blocks.conv_transpose_block + + sepconv_transpose_block + ----------------------- + .. autofunction:: akida_models.layer_blocks.sepconv_transpose_block + + tools + ----- + .. autofunction:: akida_models.layer_blocks.act_to_layer + .. autofunction:: akida_models.layer_blocks.norm_to_layer + Helpers ======= @@ -31,7 +56,7 @@ Akida models API ====================== .. autoclass:: akida_models.distiller.Distiller - :members: + .. autoclass:: akida_models.distiller.DeitDistiller .. autofunction:: akida_models.distiller.KLDistillationLoss Pruning @@ -48,6 +73,15 @@ Akida models API .. autofunction:: akida_models.training.evaluate_akida_model .. autofunction:: akida_models.training.compile_model + MACS + ==== + .. autofunction:: akida_models.macs.get_flops + .. autofunction:: akida_models.macs.display_macs + + Utils + ===== + .. autofunction:: akida_models.utils.fetch_file + Model zoo =========