Load custom model

[khawar-islam]

Dear @novice03

I would like to visualize features of my trained pre-trained model? I firstly used timm pre-trained model, fine-tune it and now i want to visualize features.