Support for PyTorch pertained models ?

[vikas95]

Hi,

Thanks for sharing these excellent resources.

I wanted to ask if there is an easy fix in the model loading part and if we can extend the same attention analysis on pytorch pertained models also ?

Looking forward for your response.