huawei-noah / Efficient-AI-Backbones

Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Training Error on ViG

Divs1159 opened this issue · comments

First off, amazing work, and thank you very much for sharing the code !

I am facing a issue when training. I got the error:
image

I am using timm version 0.3.2. can you please help me

Sorry, solved the issue. It was with the folder structure.

I have another doubt. If I am using a custom dataset, is it better to use the pre-trained weights and resume training from there?

better to use the pre-trained weights