from datasets import load_dataset
backbones_on_in1k_v1 = load_dataset("george-chou/vi_backbones", split="IMAGENET1K_V1")
backbones_on_in1k_v2 = load_dataset("george-chou/vi_backbones", split="IMAGENET1K_V2")
for weights in backbones_on_in1k_v1:
print(weights)
for weights in backbones_on_in1k_v2:
print(weights)
git clone git@hf.co:datasets/george-chou/vi_backbones
https://pytorch.org/vision/main/_modules