Include checkpoints for UKBB MRI models

biomedia-mira / causal-gen

(ICML 2023) High Fidelity Image Counterfactuals with Probabilistic Causal Models

https://arxiv.org/abs/2306.15764

MIT License

55 stars 7 forks source link

Include checkpoints for UKBB MRI models #11

Open galezon opened 1 week ago

galezon commented 1 week ago

Dear Fabio,

Do you mind including your checkpoints for the model you used for the MRI data? The notebook eval_example, under the UKBB/MIMIC section seems to load a pre-trained model.

Thank You!

fabio-deep commented 6 days ago

Hi,

You can find some checkpoints on huggingface here: https://huggingface.co/spaces/mira-causality/counterfactuals/tree/main/checkpoints/m_b_v_s

Best, Fabio