Fix #173, lr cycle default 0 vs 1. Fix #177, mirror resnest weights for future stability.

Author: rwightman

timm/models/resnest.py

@@ -36,21 +36,21 @@ def _cfg(url='', **kwargs):
     'resnest26d': _cfg(
         url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/gluon_resnest26-50eb607c.pth'),
     'resnest50d': _cfg(
-        url='https://hangzh.s3.amazonaws.com/encoding/models/resnest50-528c19ca.pth'),
+        url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-resnest/resnest50-528c19ca.pth'),
     'resnest101e': _cfg(
-        url='https://hangzh.s3.amazonaws.com/encoding/models/resnest101-22405ba7.pth',
+        url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-resnest/resnest101-22405ba7.pth',
         input_size=(3, 256, 256), pool_size=(8, 8)),
     'resnest200e': _cfg(
-        url='https://hangzh.s3.amazonaws.com/encoding/models/resnest200-75117900.pth',
+        url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-resnest/resnest200-75117900.pth',
         input_size=(3, 320, 320), pool_size=(10, 10), crop_pct=0.909, interpolation='bicubic'),
     'resnest269e': _cfg(
-        url='https://hangzh.s3.amazonaws.com/encoding/models/resnest269-0cc87c48.pth',
+        url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-resnest/resnest269-0cc87c48.pth',
         input_size=(3, 416, 416), pool_size=(13, 13), crop_pct=0.928, interpolation='bicubic'),
     'resnest50d_4s2x40d': _cfg(
-        url='https://hangzh.s3.amazonaws.com/encoding/models/resnest50_fast_4s2x40d-41d14ed0.pth',
+        url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-resnest/resnest50_fast_4s2x40d-41d14ed0.pth',
         interpolation='bicubic'),
     'resnest50d_1s4x24d': _cfg(
-        url='https://hangzh.s3.amazonaws.com/encoding/models/resnest50_fast_1s4x24d-d4a4f76f.pth',
+        url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-resnest/resnest50_fast_1s4x24d-d4a4f76f.pth',
         interpolation='bicubic')
 }
 

timm/scheduler/cosine_lr.py

@@ -103,7 +103,7 @@ def get_update_values(self, num_updates: int):
     def get_cycle_length(self, cycles=0):
         if not cycles:
             cycles = self.cycle_limit
-        assert cycles > 0
+        cycles = max(1, cycles)
         if self.t_mul == 1.0:
             return self.t_initial * cycles
         else:

timm/scheduler/scheduler_factory.py

@@ -28,7 +28,7 @@ def create_scheduler(args, optimizer):
             decay_rate=args.decay_rate,
             warmup_lr_init=args.warmup_lr,
             warmup_t=args.warmup_epochs,
-            cycle_limit=getattr(args, 'lr_cycle_limit', 0),
+            cycle_limit=getattr(args, 'lr_cycle_limit', 1),
             t_in_epochs=True,
             noise_range_t=noise_range,
             noise_pct=getattr(args, 'lr_noise_pct', 0.67),
@@ -44,7 +44,7 @@ def create_scheduler(args, optimizer):
             lr_min=args.min_lr,
             warmup_lr_init=args.warmup_lr,
             warmup_t=args.warmup_epochs,
-            cycle_limit=getattr(args, 'lr_cycle_limit', 0),
+            cycle_limit=getattr(args, 'lr_cycle_limit', 1),
             t_in_epochs=True,
             noise_range_t=noise_range,
             noise_pct=getattr(args, 'lr_noise_pct', 0.67),

timm/scheduler/tanh_lr.py

@@ -107,7 +107,7 @@ def get_update_values(self, num_updates: int):
     def get_cycle_length(self, cycles=0):
         if not cycles:
             cycles = self.cycle_limit
-        assert cycles > 0
+        cycles = max(1, cycles)
         if self.t_mul == 1.0:
             return self.t_initial * cycles
         else: