| { | |
| "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/resnet_batch512", | |
| "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": 512, | |
| "meta_hidden_dim": 256, | |
| "meta_num_layers": 3, | |
| "grad_accum_steps": 1, | |
| "atomagined": null, | |
| "balance_sources": false | |
| } |