cmmp-resnet18-256 / config.json
cgeorgiaw's picture
cgeorgiaw HF Staff
Update: retrained on the new 7,330-image CMMP dataset (ResNet-18, crop 256, meta 256x3, batch 512). Top-1 0.844, mean linear-probe R^2 0.691.
4525296 verified
raw
history blame contribute delete
518 Bytes
{
"h5_path": "/fsx/georgia_channing/microbed/CMMP-dataset/CMMP_dataset.h5",
"batch_size": 64,
"epochs": 1000,
"lr": 0.0001,
"embed_dim": 128,
"meta_dim": 7,
"save_dir": "runs/final/backbone_resnet18",
"num_workers": 4,
"loss_fn": "clip",
"mre_sigma_temp": 1.0,
"mre_reduction": "trimmed_mean",
"snn_meta_sim": "cosine",
"snn_meta_tau": 0.1,
"image_encoder": "resnet18",
"crop_size": 256,
"meta_hidden_dim": 256,
"meta_num_layers": 3,
"atomagined": null,
"balance_sources": false
}