I have developed a new customized CNN model. for which i had used different types of attention at different levels. How can I generate the formula to extract the number of features at each layer and also, how to calculate the flops. Is there any tool that can assist or anyone who can help me?
Maybe you can use model.summary() and other similar methods from Tesnorflow:
Thak you for your comment. i want to generate the formula mathematically. while model.summary () gives only the details of the model
1 Like