File size: 385 Bytes
e695911 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
test_file: './configs/test.csv'
# test file: image_path, label
image_res: 512
batch_size: 64
test_batch_size: 128
num_workers: 8
test_num_workers: 8
temp: 0.07
mlm_probability: 0.15
queue_size: 8192
momentum: 0.995
alpha: 0.4
lam: [1.0,0.4]
image_encoder_name: resnet152
loss_ratio: 1.0
loss_ratio_note: clip_ratio
finetune: './configs/base_checkpoint.pt'
# inference ckpt location
|