fix: checkpoint loading bug
Browse files- detector/model.py +1 -1
detector/model.py
CHANGED
@@ -298,7 +298,7 @@ class FontDetector(ptl.LightningModule):
|
|
298 |
optimizer, self.num_warmup_iters, self.num_iters
|
299 |
)
|
300 |
print("Load epoch:", self.load_epoch)
|
301 |
-
for _ in range(self.num_iters * self.load_epoch // self.num_epochs):
|
302 |
self.scheduler.step()
|
303 |
print("Current learning rate set to:", self.scheduler.get_last_lr())
|
304 |
return optimizer
|
|
|
298 |
optimizer, self.num_warmup_iters, self.num_iters
|
299 |
)
|
300 |
print("Load epoch:", self.load_epoch)
|
301 |
+
for _ in range(self.num_iters * (self.load_epoch + 1) // self.num_epochs):
|
302 |
self.scheduler.step()
|
303 |
print("Current learning rate set to:", self.scheduler.get_last_lr())
|
304 |
return optimizer
|