how to print out the parameters and FLOPs of every convolutional layer and fully connected layer of the whole neural network #87

yanx27 / PointASNL

PointASNL: Robust Point Clouds Processing using Nonlocal Neural Networks with Adaptive Sampling （CVPR 2020）

MIT License

260 stars 34 forks source link

how to print out the parameters and FLOPs of every convolutional layer and fully connected layer of the whole neural network #87 #17

Closed longmalongma closed 4 years ago

longmalongma commented 4 years ago

A question has been bothering me for a long time, I would like to ask you, do you know how to print out the parameters and FLOPs of every convolutional layer and fully connected layer of the whole neural network?looking forward to your reply.