Merge pull request #192 from mkaic/patch-1

Honor the --skip-scheduler flag

src/training/train.py

@@ -68,7 +68,9 @@ def train_one_epoch(model, data, epoch, optimizer, scaler, scheduler, args, tb_w
     end = time.time()
     for i, batch in enumerate(dataloader):
         step = num_batches_per_epoch * epoch + i
-        scheduler(step)
+
+        if not args.skip_scheduler:
+            scheduler(step)
 
         images, texts = batch
         images = images.to(device=device, non_blocking=True)