search

xinyiyu / Nudiff

Diffusion-based Data Augmentation for Nuclei Image Segmentation
Other
18 stars 2 forks source link

Inquire about GPU usage #5

Open fff-ttt opened 3 months ago

fff-ttt commented 3 months ago

Thank you for your paper and code very much!

Could you tell me what graphics card (GPU) you used for this project? I am using a RTX4090 with 24GB of memory size which would cause out of memory error.

Wish you a good day, Tao

vadori commented 1 month ago

Thank you for your paper and code very much!

Could you tell me what graphics card (GPU) you used for this project? I am using a RTX4090 with 24GB of memory size which would cause out of memory error.

Wish you a good day, Tao

Hi there! I am not the author of this repo, and I am not sure this can help you, but did you try to reduce the batch size?