added print_data_info to main

main.py

@@ -65,6 +65,10 @@ def main():
         predict_transforms=get_predict_transforms() if mode == "predict" else None
     )
     
+    segmentator.print_data_info(
+        loader_type=mode, index=0
+    )
+
     # Watch parameters & gradients of model
     if config.wandb_config.use_wandb:
         wandb.watch(segmentator._model, log="all", log_graph=True)