update phase1_seg config to fix issue 127

2023-04-17 18:06:37 +08:00
parent 3281a432bf
commit d5b1ae3b42
1 changed files with 2 additions and 3 deletions
@@ -1,4 +1,3 @@
-# Note  : *** the batch_size should be equal to the gpus number at the test phase!!! ***
 data_cfg:
  dataset_name: CASIA-B*
  dataset_root: your_path
@@ -11,9 +10,9 @@ data_cfg:
 evaluator_cfg:
  enable_float16: true
  restore_ckpt_strict: true
-  restore_hint: 30000
+  restore_hint: 25000
  save_name: Segmentation
-  eval_func: mean_iou
+  eval_func: evaluate_segmentation
  sampler:
    batch_size: 4
    sample_type: all_ordered