Using 2 channels(NTransmittance & Retardation) for more regional information...

Using 2 channels(NTransmittance & Retardation) for more regional information with sigmoid at decoder end

code/data.py

@@ -291,7 +291,7 @@ class TestDataModule(pl.LightningDataModule):
         else:
             print(f"Validation data for rank {rank}/{size} already prepared")
 
-        # Augmentations for training
+        # Augmentations for training.
         self._train_transforms = A.Compose(
             [
                 A.Affine(rotate=(-180, 180), translate_percent=(0.1, 0.1), shear=(-30, 30), cval=0, p=1),