Hannes Kuchelmeister
commited on
Commit
•
e882716
1
Parent(s):
9f1d8c8
fix error in resnet code, don*t run constructor in place
Browse files
src/models/focus_resnet_module.py
CHANGED
|
@@ -49,7 +49,7 @@ class ResNetLitModule(LightningModule):
|
|
| 49 |
self.pretrained = pretrained
|
| 50 |
|
| 51 |
if resnet_type == "ResNet":
|
| 52 |
-
resnet_constructor = models.ResNet
|
| 53 |
elif resnet_type == "resnet18":
|
| 54 |
resnet_constructor = models.resnet18
|
| 55 |
elif resnet_type == "resnet34":
|
|
|
|
| 49 |
self.pretrained = pretrained
|
| 50 |
|
| 51 |
if resnet_type == "ResNet":
|
| 52 |
+
resnet_constructor = models.ResNet
|
| 53 |
elif resnet_type == "resnet18":
|
| 54 |
resnet_constructor = models.resnet18
|
| 55 |
elif resnet_type == "resnet34":
|