From 5d0951615146cadb3101498a5ca6ff0a34724853 Mon Sep 17 00:00:00 2001
From: kaiza_hikaru <kaiza_hikaru@163.com>
Date: Mon, 3 Nov 2025 10:35:01 +0800
Subject: [PATCH] add tqdm ncol in test and stat

---
 MOAFStat.py | 2 +-
 MOAFTest.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/MOAFStat.py b/MOAFStat.py
index 33e8cb7..5b2acab 100644
--- a/MOAFStat.py
+++ b/MOAFStat.py
@@ -77,7 +77,7 @@ def process_autofocus_results(input_file, dataset_type, group_size=30):
 
     print_with_timestamp(f"开始处理数据，共{total_rows}行，{n_groups}个完整分组")
 
-    for group_idx in trange(n_groups):
+    for group_idx in trange(n_groups, ncols=180):
         start_idx = group_idx * group_size
         end_idx = (group_idx + 1) * group_size
 
diff --git a/MOAFTest.py b/MOAFTest.py
index bbb6c00..f8fda63 100644
--- a/MOAFTest.py
+++ b/MOAFTest.py
@@ -17,7 +17,7 @@ def test(model, test_loader, device, model_type, dataset_type):
     results = []
 
     with torch.no_grad():
-        for data in tqdm(test_loader, desc=f"[Test]"):
+        for data in tqdm(test_loader, desc=f"[Test]", ncols=180):
             images = data["image"].to(device, non_blocking=True)
             params = torch.stack((data["mag"], data["na"], data["rix"]), dim=1).to(device, non_blocking=True)
             outputs = model(images, params)