Script to test load each model

load-model.py

@@ -0,0 +1,175 @@
+#!/usr/bin/env python3
+
+from utils import load_model
+from torchvision.models import alexnet
+from torchvision.models import densenet121
+from torchvision.models import densenet161
+from torchvision.models import densenet169
+from torchvision.models import densenet201
+from torchvision.models import efficientnet_b0
+from torchvision.models import efficientnet_b1
+from torchvision.models import efficientnet_b2
+from torchvision.models import efficientnet_b3
+from torchvision.models import efficientnet_b4
+#from torchvision.models import inception_resnet_v2
+from torchvision.models import inception_v3
+#from torchvision.models import inception_v4
+from torchvision.models import mobilenet_v2
+from torchvision.models import mobilenet_v3_large
+from torchvision.models import mobilenet_v3_small
+from torchvision.models import resnet18
+from torchvision.models import resnet34
+from torchvision.models import resnet50
+from torchvision.models import resnet101
+from torchvision.models import resnet152
+from torchvision.models import shufflenet_v2_x1_0
+from torchvision.models import squeezenet
+from torchvision.models import vgg11
+from torchvision.models import vit_b_16
+from torchvision.models import wide_resnet50_2
+from torchvision.models import wide_resnet101_2
+
+use_gpu = True  # load weights on the gpu
+
+### BEGIN upstream OK ###
+#filename = '/srv/ml/plantnet/models/resnet18_weights_best_acc.tar'
+#model = resnet18(num_classes=1081) # 1081 classes in Pl@ntNet-300K
+### END upstream ###
+
+### BEGIN alexnet OK ###
+#filename = '/srv/ml/deepcrayon/plantnet-models/alexnet_weights_best_acc.tar'
+#model = alexnet(num_classes=1081)
+### END alexnet ###
+
+### BEGIN densenet121 UNTESTED ###
+#filename = '/srv/ml/deepcrayon/plantnet-models/densenet121_weights_best_acc.tar'
+#model = densenet121(num_classes=1081)
+### END densenet121 ###
+
+### BEGIN densenet161 UNTESTED ###
+#filename = '/srv/ml/deepcrayon/plantnet-models/densenet161_weights_best_acc.tar'
+#model = densenet161(num_classes=1081)
+### END densenet161 ###
+
+### BEGIN densenet169 UNTESTED ###
+#filename = '/srv/ml/deepcrayon/plantnet-models/densenet169_weights_best_acc.tar'
+#model = densenet169(num_classes=1081)
+### END densenet169 ###
+
+### BEGIN densenet201 UNTESTED ###
+#filename = '/srv/ml/deepcrayon/plantnet-models/densenet201_weights_best_acc.tar'
+#model = densenet201(num_classes=1081)
+### END densenet201 ###
+
+### BEGIN efficientnet_b0 FAIL ###
+#filename = '/srv/ml/deepcrayon/plantnet-models/efficientnet_b0_weights_best_acc.tar'
+#model = efficientnet_b0(num_classes=1081)
+### END efficientnet_b0 ###
+
+### BEGIN efficientnet_b1 FAIL ###
+#filename = '/srv/ml/deepcrayon/plantnet-models/efficientnet_b1_weights_best_acc.tar'
+#model = efficientnet_b1(num_classes=1081)
+### END efficientnet_b1 ###
+
+### BEGIN efficientnet_b2 FAIL ###
+#filename = '/srv/ml/deepcrayon/plantnet-models/efficientnet_b2_weights_best_acc.tar'
+#model = efficientnet_b2(num_classes=1081)
+### END efficientnet_b2 ###
+
+### BEGIN efficientnet_b3 FAIL ###
+#filename = '/srv/ml/deepcrayon/plantnet-models/efficientnet_b3_weights_best_acc.tar'
+#model = efficientnet_b3(num_classes=1081)
+### END efficientnet_b3 ###
+
+### BEGIN efficientnet_b4 FAIL ###
+#filename = '/srv/ml/deepcrayon/plantnet-models/efficientnet_b4_weights_best_acc.tar'
+#model = efficientnet_b4(num_classes=1081)
+### END efficientnet_b4 ###
+
+### BEGIN inception_resnet_v2 FAIL no module import ###
+#filename = '/srv/ml/deepcrayon/plantnet-models/inception_resnet_v2_weights_best_acc.tar'
+#model = inception_resnet_v2(num_classes=1081)
+### END inception_resnet_v2 ###
+
+### BEGIN inception_v3 FAIL no train ###
+#filename = '/srv/ml/deepcrayon/plantnet-models/inception_v3_weights_best_acc.tar'
+#model = inception_v3(num_classes=1081)
+### END inception_v3 ###
+
+### BEGIN inception_v4 FAIL no module import ###
+#filename = '/srv/ml/deepcrayon/plantnet-models/inception_v4_weights_best_acc.tar'
+#model = inception_v4(num_classes=1081)
+### END inception_v4 ###
+
+### BEGIN mobilenet_v2 UNTESTED ###
+#filename = '/srv/ml/deepcrayon/plantnet-models/mobilenet_v2_weights_best_acc.tar'
+#model = mobilenet_v2(num_classes=1081)
+### END mobilenet_v2 ###
+
+### BEGIN mobilenet_v3_large OK ###
+#filename = '/srv/ml/deepcrayon/plantnet-models/mobilenet_v3_large_weights_best_acc.tar'
+#model = mobilenet_v3_large(num_classes=1081)
+### END mobilenet_v3_large ###
+
+### BEGIN mobilenet_v3_small OK ###
+#filename = '/srv/ml/deepcrayon/plantnet-models/mobilenet_v3_small_weights_best_acc.tar'
+#model = mobilenet_v3_small(num_classes=1081)
+### END mobilenet_v3_small ###
+
+### BEGIN resnet18 OK ###
+#filename = '/srv/ml/deepcrayon/plantnet-models/resnet18_weights_best_acc.tar'
+#model = resnet18(num_classes=1081)
+### END resnet18 ###
+
+### BEGIN resnet34 UNTESTED ###
+#filename = '/srv/ml/deepcrayon/plantnet-models/resnet34_weights_best_acc.tar'
+#model = resnet34(num_classes=1081)
+### END resnet34 ###
+
+### BEGIN resnet50 OK ###
+#filename = '/srv/ml/deepcrayon/plantnet-models/resnet50_weights_best_acc.tar'
+#model = resnet50(num_classes=1081)
+### END resnet50 ###
+
+### BEGIN resnet101 UNTESTED ###
+#filename = '/srv/ml/deepcrayon/plantnet-models/resnet101_weights_best_acc.tar'
+#model = resnet101(num_classes=1081)
+### END resnet101 ###
+
+### BEGIN resnet152 UNTESTED ###
+#filename = '/srv/ml/deepcrayon/plantnet-models/resnet152_weights_best_acc.tar'
+#model = resnet152(num_classes=1081)
+### END resnet152 ###
+
+### BEGIN shufflenet_v2_x1_0 OK ###
+#filename = '/srv/ml/deepcrayon/plantnet-models/shufflenet_v2_x1_0_weights_best_acc.tar'
+#model = shufflenet_v2_x1_0(num_classes=1081)
+### END shufflenet_v2_x1_0 ###
+
+### BEGIN squeeznet UNTESTED ###
+#filename = '/srv/ml/deepcrayon/plantnet-models/squeeznet_weights_best_acc.tar'
+#model = squeeznet(num_classes=1081)
+### END squeeznet ###
+
+### BEGIN vgg11 UNTESTED ###
+#filename = '/srv/ml/deepcrayon/plantnet-models/vgg11_weights_best_acc.tar'
+#model = vgg11(num_classes=1081)
+### END vgg11 ###
+
+### BEGIN vit_b_16 FAIL ###
+#filename = '/srv/ml/deepcrayon/plantnet-models/vit_b_16_weights_best_acc.tar'
+#model = vit_b_16(num_classes=1081)
+### END vit ###
+
+### BEGIN wide_resnet50_2 UNTESTED ###
+#filename = '/srv/ml/deepcrayon/plantnet-models/wide_resnet50_2_weights_best_acc.tar'
+#model = wide_resnet50_2(num_classes=1081)
+### END wide_resnet50_2 ###
+
+### BEGIN wide_resnet101_2 UNTESTED ###
+#filename = '/srv/ml/deepcrayon/plantnet-models/wide_resnet101_2_weights_best_acc.tar'
+#model = wide_resnet101_2(num_classes=1081)
+### END wide_resnet101_2 ###
+
+load_model(model, filename=filename, use_gpu=use_gpu)
+