Unknow architecture: vit_small when running 1-Hierarchical-Pretraining/main_dino4k.py

[eran88]

It seems like the code doesn't recognize the "vit_small" architecture, even through it's the default architecture

[Richarizardd]

Hi @eran88 - I edited the default architecture to be vit_xs. In can also change the import statements to import your desired architecture.