config/experiment.yaml

# config.yaml

exp_name: exp1-2
wandb: 90788c79e1500570b08e5acf283e17df7e0c54b2
root: "${oc.env:DATA_ROOT}"
overfit: False
batch_size: 4
num_workers: 4
img_size: 224
rationale_type: 0 # 0 - only rationale, 1 - randomly add entity desc, 2 - add entity desc
val_rationale_type: 0
hide_true_bbox: 8 # clues and inferences selected randomly
widescreen_processing: 1 # 0 - no widescreen, 1 - widescreen
h_flip: False
ema_decay: 0.9999

clip_model: 'ViT-B/16' # 'RN101' 'RN50x4''RN50x16' 'RN50x64' 'ViT-L/14@336px' 'ViT-B/32'
num_layers: 3
dim_hidden: 512


warmup: 1000
init_from: ''
lr: .00001
n_epochs: 15
save_every: 0
early_stop: 5
val_stat: 'loss'
device: 'cuda'
use_multi: False
local_rank: 0

hydra:
  run:
    dir: ./results/${exp_name}
  output_subdir: ./ # directory for saving the yaml configs
  job:
    config:
      override_dirname:
        exclude_keys:
          - exp.name