search

Rubics-Xuan / TransBTS

This repo provides the official code for : 1) TransBTS: Multimodal Brain Tumor Segmentation Using Transformer (https://arxiv.org/abs/2103.04430) , accepted by MICCAI2021. 2) TransBTSV2: Towards Better and More Efficient Volumetric Segmentation of Medical Images(https://arxiv.org/abs/2201.12785).
Apache License 2.0
388 stars 81 forks source link

CUDA out of memory #45

Open justinzhang528 opened 1 year ago

justinzhang528 commented 1 year ago

進行訓練的時候出現=》RuntimeError: CUDA out of memory. Tried to allocate 128.00 MiB (GPU 0; 10.92 GiB total capacity; 10.17 GiB already allocated; 99.81 MiB free; 10.20 GiB reserved in total by PyTorch) 我的GPU配置： NVIDIA GeForce GTX 1080 Ti 11264 MB 我有嘗試把batch_sizej降低至1，還是未能解決，不知是否還有其他方法可以解決問題

karelbecerra commented 1 year ago

first try with batch size 1, if it doesn't work try reducing dataset size

nchzzDFTBA commented 9 months ago

现在解决了吗

chjwhere commented 9 months ago

你好，请问你解决了吗

justinzhang528 commented 9 months ago

沒有解決喔，後來選擇放棄，直接用其它模型

chjwhere commented 9 months ago

你好，我也遇到相关问题无法解决。请问你后来使用的是什么模型，能否给个参考。谢谢！