not resuming scheduler

scripts/main.py

@@ -151,13 +151,16 @@ def main():
 
     if args.resume and check_isfile(args.resume):
         args.start_epoch = resume_from_checkpoint(args.resume, model, scheduler=scheduler, optimizer=optimizer)
-    
+
     # lr changing is BUG-gy!
     if args.lr != optimizer.param_groups[0]['lr']:
         old_lr = optimizer.param_groups[0]['lr']
         change_lr_to(optimizer,args.lr)
         print('Changed optimzer lr from {} to {}.'.format(old_lr,args.lr))
 
+    # NOTE: not resuming scheduler
+    scheduler = torchreid.optim.build_lr_scheduler(optimizer, **lr_scheduler_kwargs(args))
+
     print('Building {}-engine for {}-reid'.format(args.loss, args.app))
     engine = build_engine(args, datamanager, model, optimizer, scheduler, experiment)
 

torchreid/models/osnet.py

@@ -316,7 +316,7 @@ def forward(self, x):
 def osnet_custom(num_classes=1000, loss='softmax', **kwargs):
     # custom osnet
     return OSNet(num_classes, blocks=[OSBlock, OSBlock, OSBlock], layers=[2, 2, 2],
-                 channels=[64, 256, 384, 512], feature_dim=1024, loss=loss, **kwargs)
+                 channels=[64, 256, 384, 512], feature_dim=2048, loss=loss, **kwargs)
 
 def osnet_x1_0(num_classes=1000, loss='softmax', **kwargs):
     # standard size (width x1.0)