CHE-Master / configs /example.yaml
Elfenreigen's picture
Upload 3 files
e695911 verified
raw
history blame contribute delete
385 Bytes
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