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

ViG训练自己的数据集 #142

Closed ModeSky closed 2 years ago

ModeSky commented 2 years ago

请问使用VIG训练自己的数据集，训练集应该如何构造即数据集的目录应包含那些文件夹

iamhankai commented 2 years ago

之前怎么训练的，把backbone换成ViG就行了