yjh0410 2 年之前
父節點
當前提交
b0147c0482
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      dataset/scripts/data_to_h5py.py

+ 1 - 1
dataset/scripts/data_to_h5py.py

@@ -43,7 +43,7 @@ print('Data length: ', len(dataset))
 image_dict = dict()
 dataset_size = len(dataset)
 for i in range(len(dataset)):
-    if i % 1000 == 0:
+    if i % 5000 == 0:
         print("[{} / {}]".format(i, dataset_size))
     # load an image
     image, image_id = dataset.pull_image(i)