there isn’t out of the box, but it seems valuable.
We have inception via: https://github.com/pytorch/vision/blob/master/torchvision/models/inception.py 996
A script like the other one needs to be made.