From 7812f2976f657590f5cf76b18caa0dfef7d1d241 Mon Sep 17 00:00:00 2001
From: laynholt <ne.free532@gmail.com>
Date: Fri, 9 May 2025 18:18:32 +0000
Subject: [PATCH] added print_data_info to main

---
 main.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/main.py b/main.py
index 136b013..3bf82ed 100644
--- a/main.py
+++ b/main.py
@@ -64,6 +64,10 @@ def main():
         test_transforms=get_test_transforms() if mode in ("train", "test") else None,
         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: