search

huawei-noah / Efficient-AI-Backbones

Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.
4.07k stars 708 forks source link

About vig #103

Closed chunxt closed 2 years ago

chunxt commented 2 years ago

I am a new learner for deep learning, but I want to know how to load the pretrained model you offered. Because it's format is '.pkl', I have no idea about it. And is it same as '.pth'?

iamhankai commented 2 years ago

We provide .pth.tar files which are saved in PyTorch format.

chunxt commented 2 years ago

We provide .pth.tar files which are saved in PyTorch format.

thanks. I solved it by loading the whole '.pth.tar'.