search

jeya-maria-jose / Medical-Transformer

Official Pytorch Code for "Medical Transformer: Gated Axial-Attention for Medical Image Segmentation" - MICCAI 2021
MIT License
791 stars 176 forks source link

About CUDA out of memory #86

Closed louan1998 closed 8 months ago

louan1998 commented 1 year ago

这个模型跑起来到底要多少显存？我这里有两张40G的A100居然还爆显存

nightrain-vampire commented 1 year ago

Perhaps changing the scale of the image to a smaller size would help?

louan1998 commented 1 year ago

Perhaps changing the scale of the image to a smaller size would help?

yes! I change the training image size. The first time I use the 1000 1000 size picture, then I changed it to 256 256, and it worked!