GPU requirements in README

[tomjf]

Hello, thank you for sharing the code.

What size GPU was used in training? I couldn't find this in the paper. I am running into OOM issues after 100 epochs on a 24GB RTX A5000. It would be helpful to state the GPU requirements in the README, especially if they are large. Or perhaps I have made a mistake.

[seannz]

Hello Tom,

Thanks for your interest in our work — we trained slightly different models for adult brain SVR and fetal. IIRC both were trained on an RTX 8000 with 48gb of ram. Since the publication of the paper, we have made some architectural tweaks to our models and are in the process of training smaller and more accurate models right now, but in the mean time please feel free to train a smaller model listed in flow_SNet4.py. Let me know if you have any questions
—Sean

[tomjf]

Thanks Sean,
I tried one of the smaller models and that works. Thanks again. Is the splat-slice U-Net in the paper flow_SNet3d0_192?