feat: try resnet 50
Browse files
train.py
CHANGED
|
@@ -76,7 +76,7 @@ trainer = ptl.Trainer(
|
|
| 76 |
deterministic=True,
|
| 77 |
)
|
| 78 |
|
| 79 |
-
model =
|
| 80 |
|
| 81 |
detector = FontDetector(
|
| 82 |
model=model,
|
|
|
|
| 76 |
deterministic=True,
|
| 77 |
)
|
| 78 |
|
| 79 |
+
model = ResNet50Regressor(regression_use_tanh=regression_use_tanh)
|
| 80 |
|
| 81 |
detector = FontDetector(
|
| 82 |
model=model,
|