crosstyan/OpenGait
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update phase1_seg config to fix issue 127

Browse Source
This commit is contained in:
darkliang
2023-04-17 18:06:37 +08:00
parent 3281a432bf
commit d5b1ae3b42
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/gaitedge/phase1_seg.yaml
+2 -3
View File
@@ -1,4 +1,3 @@
# Note : *** the batch_size should be equal to the gpus number at the test phase!!! ***
data_cfg: data_cfg:
dataset_name: CASIA-B* dataset_name: CASIA-B*
dataset_root: your_path dataset_root: your_path
@@ -11,9 +10,9 @@ data_cfg:
evaluator_cfg: evaluator_cfg:
enable_float16: true enable_float16: true
restore_ckpt_strict: true restore_ckpt_strict: true
restore_hint: 30000 restore_hint: 25000
save_name: Segmentation save_name: Segmentation
eval_func: mean_iou eval_func: evaluate_segmentation
sampler: sampler:
batch_size: 4 batch_size: 4
sample_type: all_ordered sample_type: all_ordered