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

On the Number of GPUs #17

Closed Li-bingjie48 closed 3 years ago

Li-bingjie48 commented 3 years ago

Hello, I see that you use four GPUs to run the code. Is there only one GPU in my computer that can run through? Look forward to your reply.

gshaowei6 commented 3 years ago

可以,但是如果使用一块GPU来训练的话,当batchsize设置为2的时候,显存已经需要24g了,所以你如果仅有一块GPU的话,估计得是3090,代码只用将分布式改为非分布式训练即可

Li-bingjie48 commented 3 years ago

谢谢你的解答 非常感谢

------------------ 原始邮件 ------------------ 发件人: "Wenxuan-1119/TransBTS" @.>; 发送时间: 2021年6月9日(星期三) 晚上9:22 @.>; @.**@.>; 主题: Re: [Wenxuan-1119/TransBTS] On the Number of GPUs (#17)

可以,但是如果使用一块GPU来训练的话,当batchsize设置为2的时候,显存已经需要24g了,所以你如果仅有一块GPU的话,估计得是3090,代码只用将分布式改为非分布式训练即可

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.

Lnan1615 commented 2 years ago

我也只有一块3090,但是我不太清楚要怎么改非分布式,有改好的的代码让我学习一下吗?

jianglanlele commented 2 years ago

Hello, I see that you use four GPUs to run the code. Is there only one GPU in my computer that can run through? Look forward to your reply.

@Li-bingjie48 你好,你实现单卡运行的代码了嘛,想参考一下

Lnan1615 commented 2 years ago

我改了代码,也跑通了,但是结果的波动超级大,不收敛的样子,应该是我哪里改的有问题。你改出来的代码也是这个样子吗?

jianglanlele commented 2 years ago

我改了代码,也跑通了,但是结果的波动超级大,不收敛的样子,应该是我哪里改的有问题。你改出来的代码也是这个样子吗?

能让我参考下嘛,不太会改代码,感谢,qq邮箱 3057748270@qq.com

chjwhere commented 9 months ago

我改了代码,也跑通了,但是结果的波动超级大,不收敛的样子,应该是我哪里改的有问题。你改出来的代码也是这个样子吗?

你好,请问你们是如何将分布式训练更改为非分布式训练?我也仅有一张3090,但是不太懂得如何更改代码以训练。希望能得到你们的回复,万分感谢!1641377171@qq.com

chjwhere commented 9 months ago

我改了代码,也跑通了,但是结果的波动超级大,不收敛的样子,应该是我哪里改的有问题。你改出来的代码也是这个样子吗?

能让我参考下嘛,不太会改代码,感谢,qq邮箱 3057748270@qq.com

你好,请问你们是如何将分布式训练更改为非分布式训练?我也仅有一张3090,但是不太懂得如何更改代码以训练。希望能得到你们的回复,万分感谢!1641377171@qq.com

Taocunguo commented 6 months ago

我改了代码,也跑通了,但是结果的波动超级大,不收敛的样子,应该是我哪里改的有问题。你改出来的代码也是这个样子吗?

你好,我也不太会改代码,请问你可以也发一份改好的给我吗,万分谢谢。1460535844@qq.com