Is there any way to compute the number of trainable parameters from code in a deep CNN network?

9 次查看（过去 30 天）
krishna Chauhan
krishna Chauhan 2022-12-20
回答： Harsh 2025-2-28
Hello
I am using a six layer compact CNN model for classification after intantiating the layers and training data to trainNetwork(). I want to calculate the number of trainable parameters in this network. Something similar to the below in pytorch:
model=net()
for parameter in model.parameters():
print(parameter)
Is there any way avaible in MATLAB?

请先登录，再回答此问题。

回答（1 个）

Harsh
Harsh 2025-2-28
Hi @krishna Chauhan,
You can calculate the number of trainable parameters in your Convolutional Neural Network (CNN) model using the “analyzeNetwork” function, which provides a detailed layer-by-layer breakdown, including the count of learnable parameters for each layer.
Please refer to the following documentation for more information regarding “analyzeNetwork” function - https://www.mathworks.com/help/deeplearning/ref/analyzenetwork.html

类别

在 Help Center 和 File Exchange 中查找有关 Deep Learning Toolbox 的更多信息

标签

产品

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by