ACE-LoRA / config.json
aydnarda's picture
Push model using huggingface_hub.
f62ac4d verified
raw
history blame contribute delete
725 Bytes
{
"alpha": 1,
"apply_gnn_encoders": "both",
"base_model_name": "hf-hub:microsoft/BiomedCLIP-PubMedBERT_256-vit_base_patch16_224",
"batch_size": 64,
"context_length": 256,
"csv_path": "./mimic_cxr.csv",
"dropout_rate": 0.1,
"eval": true,
"feature_dim": 512,
"filename": "ACE-LoRA_weights",
"hidden_features": 128,
"label_csv_path": "mimic-cxr-chexpert_labeled_data.csv",
"learnable_logit_scale": true,
"loss_type": "clip_loss_ace_hgnn",
"lr": 0.001,
"num_devices": 3,
"num_epochs": 40,
"output_dim": 512,
"params": [
"q",
"k",
"v"
],
"position": "all",
"r": 4,
"rank": 0,
"root_path": "",
"save_path": "./ACE-LoRA",
"seed": 1,
"topk": 5,
"use_labels": true
}