Update model_loader.py

model_loader.py

@@ -19,26 +19,6 @@ def build_model(nclasses: int = 2, mode: str = None, segment_model: str = None):
 
     return net
 
-def build_SurgFM(nclasses: int = 2, pretrained: bool = True, pretrained_weights = None):
-
-
-    #net of ConvNext
-    net = torchvision.models.convnext_large(weights='DEFAULT')
-    input_emdim = net.classifier[2].in_features
-    net.classifier[2] = nn.Identity()
-    
-    if os.path.isfile(pretrained_weights):
-        state_dict = torch.load(pretrained_weights, map_location="cpu")
-        state_dict = state_dict['teacher']
-
-        # remove `backbone.` prefix induced by multicrop wrapper
-        state_dict = {k.replace("backbone.", ""): v for k, v in state_dict.items() if k.startswith('backbone.')}
-        msg = net.load_state_dict(state_dict, strict=False)
-        print(msg, input_emdim)
-
-    net.cuda()
-
-    return net
 
 
 net = build_model(nclasses=num_classes, mode='classify')