Update train_CDMA_Plus.py

HiLab-git · Apr 17, 2024 · 6ba8814 · 6ba8814
1 parent 28980ed
commit 6ba8814
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/train_CDMA_Plus.py b/train_CDMA_Plus.py
@@ -354,11 +354,9 @@ def setup_seed(seed):
     l.info(f'labeled:{labeled_num},unlabeled:{all_data_num-labeled_num}')
     print(f'labeled:{labeled_num},unlabeled:{all_data_num-labeled_num}')
 
-    val_pos, val_neg = get_files_redistribution(val_data_root)
-    val_files = val_pos
+    val_files = get_files(val_data_root)
 
-    test_pos, test_neg = get_files_redistribution(test_data_root)
-    test_files = test_pos + test_neg
+    test_files = get_files(test_data_root)
 
     l.info(f'training files:{all_data_num}, valid files:{len(val_files)}')
     print(f'training files:{all_data_num}, valid files:{len(val_files)}')