From db68ff34806175684ffa9df98f2e69503d78ee58 Mon Sep 17 00:00:00 2001 From: PiggyJerry Date: Wed, 27 Jul 2022 23:54:20 +0400 Subject: [PATCH] update data_loader_cache.py --- IS-Net/data_loader_cache.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/IS-Net/data_loader_cache.py b/IS-Net/data_loader_cache.py index 6ba9dd5..75dfb51 100644 --- a/IS-Net/data_loader_cache.py +++ b/IS-Net/data_loader_cache.py @@ -76,7 +76,7 @@ def create_dataloaders(name_im_gt_list, cache_size=[], cache_boost=True, my_tran gos_dataloaders = [] gos_datasets = [] - # if(mode=="train"): + if(len(name_im_gt_list)==0): return gos_dataloaders, gos_datasets