search

OliverRensu / Shunted-Transformer

209 stars 20 forks source link

About object detection #3

Open Hzj199 opened 2 years ago

Hzj199 commented 2 years ago

I used the shunted transformer in mmdetection and found that the GPU memory usage is very high. Can you release the code for detection and segmentation?

OliverRensu commented 2 years ago

please use torch.utils.checkpoint and we train on 8 a6000 48g gpu. We will consider releasing the code

在 2022年4月10日星期日，Hzj199 @.***> 写道：

I used the shunted transformer in mmdetection and found that the GPU memory usage is very high. Can you release the code for detection and segmentation?

— Reply to this email directly, view it on GitHub https://github.com/OliverRensu/Shunted-Transformer/issues/3, or unsubscribe https://github.com/notifications/unsubscribe-auth/ANP7CCFLKPW7TFLXEAOVYALVEI2FFANCNFSM5TADQQ5A . You are receiving this because you are subscribed to this thread.Message ID: @.***>