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 2 weeks ago

galezon commented 2 weeks 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 2 weeks 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