LabelAnything/test_validation.yaml at main · Fredbcx/LabelAnything · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
dataloader:
  num_workers: 0
  possible_batch_example_nums:
  - - 1
    - 4
  prompt_types:
  - point
  - bbox
  - mask
  val_possible_batch_example_nums:
  - - 1
    - 1
  val_prompt_types:
  - mask
dataset:
  common:
    add_box_noise: true
    do_subsample: true
    image_size: 1024
    max_points_annotations: 50
    max_points_per_annotation: 10
  datasets:
    coco20i:
      img_dir: /ext/stalla/LabelAnything/images/train2017
      instances_path: data/annotations/instances_train2014.json
      n_folds: 4
      name: coco
      remove_small_annotations: true
      split: train
      val_fold_idx: 0
    val_coco20i:
      add_box_noise: false
      do_subsample: false
      img_dir: /ext/stalla/LabelAnything/images/train2017
      instances_path: data/annotations/instances_val2014.json
      n_folds: 4
      n_shots: min
      n_ways: max
      name: coco
      split: val
      val_fold_idx: 0
experiment:
  continue_with_errors: false
  direction: null
  excluded_files: ''
  group: Test
  logger: null
  max_parallel_runs: 1
  n_trials: null
  name: LabelAnything
  resume: false
  resume_last: false
  search: grid
  start_from_grid: 0
  start_from_run: 0
  timestamp: '20240224_125600'
  tracking_dir: ''
  uuid: null
logger:
  experiment_save_delta: null
  log_frequency: 1
  tags:
  - Test
  tmp_dir: tmp
  train_image_log_frequency: 5
  val_image_log_frequency: 5
  wandb:
    entity: cilabuniba
    offline: false
model:
  name: dummy
train_params:
  accumulate_substitution: false
  check_nan: 1
  compile: false
  freeze_backbone: true
  initial_lr: 1.0e-05
  loss:
    class_weighting: true
    components:
      dice:
        weight: 0.025
      focal:
        weight: 0.5
      prompt_contrastive:
        weight: 0.475
  max_epochs: 4
  num_points: 1
  optimizer: AdamW
  scheduler:
    num_warmup_steps: 1000
    step_moment: batch
    type: cosine
  seed: 42
  substitute: false
  validation_reruns: 2
  watch_metric: loss