search

yitu-opensource / T2T-ViT

ICCV2021, Tokens-to-Token ViT: Training Vision Transformers from Scratch on ImageNet
Other
1.14k stars 177 forks source link

How to register new models? #57

Closed lucastononrodrigues closed 3 years ago

lucastononrodrigues commented 3 years ago

Hello, I would like to register new models and run your code on them. How exactly should I do it? I tried to create new functions with @register_model and create new _cfgs but it didn't work out and accuses that the model is unknown.

raise RuntimeError('Unknown model (%s)' % model_name)

Thanks