Spaces:
Runtime error
Runtime error
File size: 736 Bytes
b828b8f 32d3d37 b828b8f 5d3ea78 d49cf43 32d3d37 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
#######################################################
# This file stores all the models used in the project.#
#######################################################
# Import all models from torchvision.models
from torchvision.models import resnet50
from torchvision.models import resnet18
from torchvision.models import squeezenet1_0
from torchvision.models import vgg16
from torchvision.models import alexnet
from torchvision.models import densenet121
from torchvision.models import googlenet
from torchvision.models import inception_v3
from torchvision.models import mobilenet_v2
from torchvision.models import mobilenet_v3_small
from torchvision.models import mobilenet_v3_large
from torchvision.models import shufflenet_v2_x0_5
|