LabelAnything/test_parameters_pascal.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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
wandb_version: 1

_wandb:
  desc: null
  value:
    python_version: 3.11.0
    cli_version: 0.16.3
    framework: huggingface
    huggingface_version: 4.37.2
    is_jupyter_run: false
    is_kaggle_kernel: false
    start_time: 1708451409.63625
    t:
      1:
      - 1
      - 11
      - 35
      - 41
      - 49
      - 55
      - 63
      - 71
      2:
      - 1
      - 11
      - 35
      - 41
      - 49
      - 55
      - 63
      - 71
      3:
      - 3
      - 4
      - 7
      - 18
      - 23
      - 37
      - 42
      4: 3.11.0
      5: 0.16.3
      6: 4.37.2
      8:
      - 5
      13: linux-x86_64
    m:
    - 1: train/step
      6:
      - 3
    - 1: validate/step
      6:
      - 3
dataloader:
  desc: null
  value:
    num_workers: 32
    possible_batch_example_nums:
    - - 2
      - 8
    - - 4
      - 4
    - - 8
      - 2
    - - 16
      - 1
    prefetch_factor: 2
    prompt_types:
    - point
    - bbox
    - mask
    val_possible_batch_example_nums:
    - - 2
      - 1
    val_prompt_types:
    - mask
dataset:
  desc: null
  value:
    common:
      add_box_noise: true
      do_subsample: true
      load_gts: false
      max_points_annotations: 50
      max_points_per_annotation: 10
    datasets:
      coco20i:
        emb_dir: /leonardo_scratch/large/userexternal/nfanelli/embeddings
        img_dir: /leonardo_scratch/large/userexternal/nfanelli/train_val_2017
        instances_path: /leonardo_scratch/large/userexternal/nfanelli/annotations/instances_train2014.json
        n_folds: 4
        name: coco
        split: train
        val_fold_idx: 0
      val_coco20i:
        add_box_noise: false
        do_subsample: false
        emb_dir: /leonardo_scratch/large/userexternal/nfanelli/embeddings
        img_dir: /leonardo_scratch/large/userexternal/nfanelli/train_val_2017
        instances_path: /leonardo_scratch/large/userexternal/nfanelli/annotations/instances_val2014.json
        n_folds: 4
        n_shots: 1
        name: coco
        split: val
        val_fold_idx: 0
experiment:
  desc: null
  value:
    continue_with_errors: false
    direction: null
    excluded_files: ''
    group: Coco20iBigEmbedding
    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: '20240220_180401'
    tracking_dir: ''
    uuid: 91426e25
logger:
  desc: null
  value:
    experiment_save_delta: null
    log_frequency: 10
    tmp_dir: /leonardo_work/IscrC_PENELOPE/tmp
    train_image_log_frequency: 100000
    val_image_log_frequency: 100
    wandb:
      entity: cilabuniba
      offline: false
      offline_directory: /leonardo_work/IscrC_PENELOPE/offline
model:
  desc: null
  value:
    class_attention: true
    embed_dim: 512
    fusion_transformer: TwoWayTransformer
    name: lam_no_vit
    spatial_convs: 3
train_params:
  desc: null
  value:
    accumulate_substitution: true
    check_nan: 1
    compile: false
    initial_lr: 5.0e-05
    loss:
      class_weighting: true
      components:
        dice:
          weight: 0.025
        focal:
          weight: 0.5
        prompt_contrastive:
          weight: 0.475
    max_epochs: 50
    num_points: 1
    optimizer: AdamW
    scheduler:
      num_warmup_steps: 1000
      step_moment: batch
      type: constant_with_warmup
    seed: 42
    substitute: true
    validation_reruns: 5