Fix Model Load Error.

src/models/mobilenet.py

@@ -7,9 +7,9 @@ class MobileNetV3Large(nn.Module):
 
     def __init__(self, ckpt, num_classes) -> None:
         super().__init__()
-        self.model = torchvision.models.mobilenet_v3_large(pretrained=True)
+        self.model = torchvision.models.mobilenet_v3_large(pretrained=False)
         self.model.classifier[3] = nn.Linear(self.model.classifier[3].in_features, num_classes)
-        self.model.load_state_dict(torch.load(ckpt))
+        self.model.load_state_dict(torch.load(ckpt, map_location=torch.device('cpu')))
         self.model.to('cpu')
         self.model.eval()