From d5b1ae3b42aa8b56c88290bb18758121862f256d Mon Sep 17 00:00:00 2001 From: darkliang <12132342@mail.sustech.edu.cn> Date: Mon, 17 Apr 2023 18:06:37 +0800 Subject: [PATCH] update phase1_seg config to fix issue 127 --- configs/gaitedge/phase1_seg.yaml | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/configs/gaitedge/phase1_seg.yaml b/configs/gaitedge/phase1_seg.yaml index 967f142..d193628 100644 --- a/configs/gaitedge/phase1_seg.yaml +++ b/configs/gaitedge/phase1_seg.yaml @@ -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