This repository was archived by the owner on Oct 16, 2023. It is now read-only.
This repository was archived by the owner on Oct 16, 2023. It is now read-only.
Where is model_zoo module? #43
Description
🐛 Describe the bug
I try to train vision transformer following instructions in README.md. However, it throws an error in imagenet1k/train.py that there is no module named model_zoo. Corresponding code is from model_zoo.vit import vit_small_patch16_224. I tried to find this module in all repos in hpcaitech organization and required python wheels, but nothing was found.
My training script is DATA=../dataset/tfrecord torchrun --nproc_per_node=8 train.py --config=configs/vit_vanilla.py , which is executed in imagenet1k folder.
Environment
CUDA 11.3
Torch 1.12.1
Torchvision 0.13.1