Hi friends in Pytorch community,
I want to find out the inference time of each (conv, bn, relu) layer in a pytorch model (like resnet101 implemented in torchvision package), is there any way to do this? I guess maybe register_buffer 3  is suitable for this task, but I don’t know how to figure it out. Does anyone has any idea? Thanks in advance