hubconf.py

from torch_hub import (pdiscoformer_cub_k_8, pdiscoformer_cub_k_4, pdiscoformer_cub_k_16,
                       pdiscoformer_flowers_k_2, pdiscoformer_flowers_k_4, pdiscoformer_flowers_k_8,
                       pdiscoformer_pimagenet_k_25, pdiscoformer_pimagenet_k_50, pdiscoformer_pimagenet_k_8,
                       pdiscoformer_pimagenet_seg_k_8, pdiscoformer_pimagenet_seg_k_16, pdiscoformer_pimagenet_seg_k_25,
                       pdiscoformer_pimagenet_seg_k_41, pdiscoformer_pimagenet_seg_k_50,
                       pdiscoformer_nabirds_k_4, pdiscoformer_nabirds_k_8, pdiscoformer_nabirds_k_11,
                       pdisconet_vit_nabirds_k_4, pdisconet_vit_nabirds_k_8, pdisconet_vit_nabirds_k_11,
                       pdisconet_resnet_nabirds_k_4, pdisconet_resnet_nabirds_k_8, pdisconet_resnet_nabirds_k_11)

dependencies = ['torch', 'torchvision', 'timm']