fix: `load_epoch` init for checkpoint loading
Browse files- detector/model.py +1 -1
detector/model.py
CHANGED
|
@@ -202,7 +202,7 @@ class FontDetector(ptl.LightningModule):
|
|
| 202 |
self.num_warmup_iters = num_warmup_iters
|
| 203 |
self.num_iters = num_iters
|
| 204 |
self.num_epochs = num_epochs
|
| 205 |
-
self.load_epoch =
|
| 206 |
self.font_classification_only = font_classification_only
|
| 207 |
|
| 208 |
def forward(self, x):
|
|
|
|
| 202 |
self.num_warmup_iters = num_warmup_iters
|
| 203 |
self.num_iters = num_iters
|
| 204 |
self.num_epochs = num_epochs
|
| 205 |
+
self.load_epoch = -1
|
| 206 |
self.font_classification_only = font_classification_only
|
| 207 |
|
| 208 |
def forward(self, x):
|