Spaces:

gyrojeff
/

YuzuMarker.FontDetection

Running

gyrojeff commited on Mar 24, 2023

Commit

acab651

1 Parent(s): 6ff7b63

fix: remove deprecated parameter

Files changed (1) hide show

detector/model.py CHANGED Viewed

@@ -13,7 +13,7 @@ import pytorch_lightning as ptl
 class ResNet18Regressor(nn.Module):
     def __init__(self):
         super().__init__()
-        self.model = torchvision.models.resnet18(pretrained=False)
         self.model.fc = nn.Linear(512, config.FONT_COUNT + 12)
     def forward(self, X):

 class ResNet18Regressor(nn.Module):
     def __init__(self):
         super().__init__()
+        self.model = torchvision.models.resnet18(weights=False)
         self.model.fc = nn.Linear(512, config.FONT_COUNT + 12)
     def forward(self, X):