fix default logger in cgan training and default dataset

mu_map/dataset/default.py

@@ -46,7 +46,7 @@ class MuMapDataset(Dataset):
 
         self.transform_normalization = transform_normalization
         self.transform_augmentation = transform_augmentation
-        self.logger = logger if logger is not None else get_logger()
+        self.logger = logger if logger is not None else get_logger(name=MuMapDataset.__name__)
 
         self.bed_contours_file = (
             os.path.join(dataset_dir, bed_contours_file) if bed_contours_file else None

mu_map/training/cgan.py

@@ -42,7 +42,7 @@ class cGANTraining:
 
         self.snapshot_dir = snapshot_dir
         self.snapshot_epoch = snapshot_epoch
-        self.logger = logger if logger is not None else get_logger()
+        self.logger = logger if logger is not None else get_logger(name=cGANTraining.__name__)
 
         self.params_g = params_generator
         self.params_d = params_discriminator
@@ -79,10 +79,8 @@ class cGANTraining:
                 self._store_snapshot(epoch)
 
             if self.params_d.lr_scheduler is not None:
-                self.logger.debug("Step LR scheduler of discriminator")
                 self.params_d.lr_scheduler.step()
             if self.params_g.lr_scheduler is not None:
-                self.logger.debug("Step LR scheduler of generator")
                 self.params_g.lr_scheduler.step()
         return loss_val_min