@ptrblck if you  could give an working example with a torchvision model then it could help a lot
i searched in  google,posted in github also in stackoverflow and got answer nowhere so a working example demo could be of great help,thank you anyway