[FIX] fix ReNet val_dataloader

RL-VIG · Oct 13, 2022 · ee25973 · ee25973
1 parent f002bf3
commit ee25973
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/core/data/dataloader.py b/core/data/dataloader.py
@@ -80,7 +80,7 @@ def get_dataloader(config, mode, model_type, distribute):
         use_memory=config["use_memory"],
     )
 
-    if config["dataloader_num"] == 1 or mode == "test":
+    if config["dataloader_num"] == 1 or mode in ["val", "test"]:
 
         collate_function = get_collate_function(config, trfms, mode, model_type)