Where is model_zoo module?

[GhostScreaming]

🐛 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