|
train_file: "setting/rad_graph_metric_train_local.json" |
|
valid_file: "setting/rad_graph_metric_validate_local.json" |
|
test_file: "setting/rad_graph_metric_test_local.json" |
|
label_file: "setting/landmark_observation_adj_mtx.npy" |
|
pathology_book: "PreTrain_MeDSLIP/data_file/observation explanation.json" |
|
|
|
image_res: 224 |
|
patch_size: 16 |
|
num_sentences: 12 |
|
num_tokens: 32 |
|
vision_width: 768 |
|
fea_width: 197 |
|
embed_dim: 256 |
|
batch_size: 64 |
|
test_batch_size: 32 |
|
temp: 0.07 |
|
mlm_probability: 0.15 |
|
queue_size: 8192 |
|
momentum: 0.995 |
|
alpha: 0.4 |
|
d_model: 256 |
|
res_base_model: "resnet50" |
|
num_queries: 75 |
|
dropout: 0.1 |
|
attribute_set_size: 2 |
|
N: 4 |
|
H: 4 |
|
no_cl: False |
|
|
|
exclude_class: False |
|
text_encoder: "emilyalsentzer/Bio_ClinicalBERT" |
|
shuffle_ratio: 0.5 |
|
optimizer: {opt: adamW, lr: 1e-4, weight_decay: 0.02} |
|
schedular: {sched: cosine, lr: 1e-4, epochs: 100, min_lr: 1e-5, decay_rate: 1, warmup_lr: 1e-5, warmup_epochs: 5, cooldown_epochs: 0} |
|
|