Merge pull request #102 from hzdr-MedImaging/fix-NoneType-scheduler

fix NoneType situation if lr_scheduler is disabled.

pytorch3dunet/unet3d/trainer.py

@@ -195,8 +195,9 @@ def train(self):
                 # adjust learning rate if necessary
                 if isinstance(self.scheduler, ReduceLROnPlateau):
                     self.scheduler.step(eval_score)
-                else:
+                elif self.scheduler is not None:
                     self.scheduler.step()
+
                 # log current learning rate in tensorboard
                 self._log_lr()
                 # remember best validation metric