Spaces:
Runtime error
Runtime error
File size: 1,123 Bytes
01e491f |
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 |
global:
name: pretrain-vision-model
phase: train
stage: pretrain-vision
workdir: workdir
seed: ~
dataset:
train: {
roots: ['data/training/MJ/MJ_train/',
'data/training/MJ/MJ_test/',
'data/training/MJ/MJ_valid/',
'data/training/ST'],
batch_size: 384
}
test: {
roots: ['data/evaluation/IIIT5k_3000',
'data/evaluation/SVT',
'data/evaluation/SVTP',
'data/evaluation/IC13_857',
'data/evaluation/IC15_1811',
'data/evaluation/CUTE80'],
batch_size: 384
}
data_aug: True
multiscales: False
num_workers: 14
training:
epochs: 8
show_iters: 50
eval_iters: 3000
save_iters: 3000
optimizer:
type: Adam
true_wd: False
wd: 0.0
bn_wd: False
clip_grad: 20
lr: 0.0001
args: {
betas: !!python/tuple [0.9, 0.999], # for default Adam
}
scheduler: {
periods: [6, 2],
gamma: 0.1,
}
model:
name: 'modules.model_vision.BaseVision'
checkpoint: ~
vision: {
loss_weight: 1.,
attention: 'position',
backbone: 'transformer',
backbone_ln: 3,
}
|