a Python 2-3 compatibility issue in lr_scheduler.py (#342)

d28845e1 · guanfuchen · Francisco Massa · 0844f577 · d28845e1
Commit d28845e1 authored Jan 14, 2019 by guanfuchen Committed by Francisco Massa Jan 14, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

lr_scheduler.py maskrcnn_benchmark/solver/lr_scheduler.py +1 -1

No files found.
--- a/maskrcnn_benchmark/solver/lr_scheduler.py
+++ b/maskrcnn_benchmark/solver/lr_scheduler.py
@@ -42,7 +42,7 @@ class WarmupMultiStepLR(torch.optim.lr_scheduler._LRScheduler):
            if self.warmup_method == "constant":
                warmup_factor = self.warmup_factor
            elif self.warmup_method == "linear":
-                alpha = self.last_epoch / self.warmup_iters
+                alpha = float(self.last_epoch) / self.warmup_iters
                warmup_factor = self.warmup_factor * (1 - alpha) + alpha
        return [
            base_lr