Update README.md

README.md

@@ -202,9 +202,15 @@ Input normalization was performed
   * VerticalFlip(p=0.5)
   * HorizontalFlip(p=0.5)
   * RandomRotate90(p=0.5)
+* Input normalization (mean=0 | std=1):
+  * norm_means: [105.08, 110.87, 101.82, 106.38, 53.26]
+  * norm_stds: [52.17, 45.38, 44, 39.69, 79.3]
 * Seed: 2022
 * Batch size: 10
+* Number of epochs : 200
+* Early stopping : patience 30 and val_loss as monitor criterium 
 * Optimizer : SGD
+* Schaeduler : mode = "min", factor = 0.5, patience = 10, cooldown = 4, min_lr = 1e-7
 * Learning rate : 0.02
 * Class Weights : 
   * 1: [1, building]
@@ -232,6 +238,12 @@ Input normalization was performed
 
 <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
 
+The FLAIR-INC_RVBIE_resnet34_unet_15cl_norm model was trained on a HPC/AI resources provided by GENCI-IDRIS (Grant 2022-A0131013803). 16 V100 GPUs were requested ( 4 nodes, 4 GPUS per node).
+
+FLAIR-INC_RVBIE_resnet34_unet_15cl_norm was obtained for num_epoch=76 with corresponding val_loss=0.56. 
+
+
+
 {{ speeds_sizes_times | default("[More Information Needed]", true)}}
 
 ## Evaluation