Merge pull request #1429 from sagarbhathwar/sagarbhathwar-patch-1

Sample config fix

recognition/ArcFace/image_iter.py

@@ -327,11 +327,11 @@ def get_face_image_iter(cfg, data_shape, path_imgrec):
             data_shape=data_shape,
             path_imgrec=path_imgrec,
             shuffle=True,
-            rand_mirror=config.data_rand_mirror,
+            rand_mirror=cfg.data_rand_mirror,
             mean=None,
-            cutoff=config.data_cutoff,
-            color_jittering=config.data_color,
-            images_filter=config.data_images_filter,
+            cutoff=cfg.data_cutoff,
+            color_jittering=cfg.data_color,
+            images_filter=cfg.data_images_filter,
         )
         train_dataiter = mx.io.PrefetchingIter(train_dataiter)
     else:

recognition/ArcFace/sample_config.py

@@ -219,5 +219,6 @@ def generate_config(_network, _dataset, _loss):
     config.network = _network
     config.dataset = _dataset
     config.num_workers = 1
+    config.fp16 = False
     if 'DMLC_NUM_WORKER' in os.environ:
         config.num_workers = int(os.environ['DMLC_NUM_WORKER'])