Spaces:
Runtime error
Runtime error
File size: 886 Bytes
f638d9c | 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 | # 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
|