search

BioMedAI-UCSC / InverseSR

[Early Accepted at MICCAI 2023] Pytorch Code of "InverseSR: 3D Brain MRI Super-Resolution Using a Latent Diffusion Model"
http://arxiv.org/abs/2308.12465
Apache License 2.0
50 stars 7 forks source link

Hardware requirements #6

Closed jakub-prokop closed 1 year ago

jakub-prokop commented 1 year ago

Hi! I'm facing constant memory issues with running the ddim pipeline on my CPU, so I am wondering, what hardware did you run your experiments on? Most importantly, what was your RAM memory size?

Thanks for all previous answers Jakub

jueqiw commented 1 year ago

You might not want to hear that, I run the experiments on a cluster with nodes with 188GB of RAM. The pretrained brain latent diffusion model we used is trained on 80 GB memory of GPU, so I guess it can be run on 80GB memory.

jakub-prokop commented 1 year ago

That explains my problems a bit :) Thanks! I close the issue.