Update model_code.py
Browse files- model_code.py +1 -1
model_code.py
CHANGED
|
@@ -348,7 +348,7 @@ class GeometricBasinClassifier(nn.Module):
|
|
| 348 |
if pretrained:
|
| 349 |
resnet = resnet18(weights=ResNet18_Weights.IMAGENET1K_V1)
|
| 350 |
else:
|
| 351 |
-
resnet = resnet18(weights=None)
|
| 352 |
|
| 353 |
# Extract feature extractor (everything except fc layer)
|
| 354 |
self.backbone = nn.Sequential(
|
|
|
|
| 348 |
if pretrained:
|
| 349 |
resnet = resnet18(weights=ResNet18_Weights.IMAGENET1K_V1)
|
| 350 |
else:
|
| 351 |
+
resnet = resnet18(weights=None) # will be running both types of train labeled
|
| 352 |
|
| 353 |
# Extract feature extractor (everything except fc layer)
|
| 354 |
self.backbone = nn.Sequential(
|