jhj0517 commited on
Commit
7036f3f
1 Parent(s): 0cb1388

Add `cb_multimask_output`

Browse files
Files changed (1) hide show
  1. configs/default_hparams.yaml +3 -2
configs/default_hparams.yaml CHANGED
@@ -1,4 +1,4 @@
1
- mask_gen_hparams:
2
  points_per_side: 64
3
  points_per_batch: 128
4
  pred_iou_thresh: 0.7
@@ -8,4 +8,5 @@ mask_gen_hparams:
8
  box_nms_thresh: 0.7
9
  crop_n_points_downscale_factor: 2
10
  min_mask_region_area: 25.0
11
- use_m2m: true
 
 
1
+ mask_hparams:
2
  points_per_side: 64
3
  points_per_batch: 128
4
  pred_iou_thresh: 0.7
 
8
  box_nms_thresh: 0.7
9
  crop_n_points_downscale_factor: 2
10
  min_mask_region_area: 25.0
11
+ use_m2m: true
12
+ multimask_output: true