microsoft

/

mri-autoencoder-v0.1

like 15

Follow

Microsoft 11.7k

Thanks! I think the easiest way to finetune would be to use the diffusers library. The notebook gives an example of how to load the weights for inference. For finetuning you will have to write your own training module. Depending on interest, we will look into open sourcing the training code at some point.

i'm pretty interested in the training code-base.

Hey, i would also be interested in the training process. Could you please share the code-basis to me?
I‘m currently doing my research in MRI-Reconstruction and therefore your work is very interesting for me:)