浏览代码

debug cache

yjh0410 2 年之前
父节点
当前提交
0e014dd99f
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      dataset/coco.py

+ 1 - 1
dataset/coco.py

@@ -102,7 +102,7 @@ class COCODataset(Dataset):
         self.cached_images = []
         self.cached_images = []
         self.cached_targets = []
         self.cached_targets = []
         dataset_size = len(self.ids)
         dataset_size = len(self.ids)
-
+        print('loading data into cache ...')
         for i in range(dataset_size):
         for i in range(dataset_size):
             if i % 5000 == 0 and i > 0:
             if i % 5000 == 0 and i > 0:
                 print("[{} / {}]".format(i, dataset_size))
                 print("[{} / {}]".format(i, dataset_size))