Open Micla-SHL opened 11 months ago
更新资源消耗: 4090 显存24G 运行bash scripts/Pretrain_Micla.sh /Micla/Project/ESTextSpotter/datasets/ datasets只包含totaltext数据集,Pretrain_Micla.sh 相比原版作了显卡8>1 train_dataset剔除剩下 totaltext ,配置batch size 2> 1, 所需显存 21G
可以把ESTextSpotter中的denoising训练部分关了可以节省显存
好的,我去尝试一下。
发件人: mxin262 @.> 发送时间: 2023年10月23日 14:56 收件人: mxin262/SwinTextSpotter @.> 抄送: Micla-SHL @.>; Author @.> 主题: Re: [mxin262/SwinTextSpotter] 关于SwinTextSpotter and ESTextSpotter 一些问题 (Issue #107)
可以把ESTextSpotter中的denoising训练部分关了可以节省显存
― Reply to this email directly, view it on GitHubhttps://github.com/mxin262/SwinTextSpotter/issues/107#issuecomment-1774543915, or unsubscribehttps://github.com/notifications/unsubscribe-auth/AKTQYIVXXYYGZXMCHHU4GYDYAYIINAVCNFSM6AAAAAA55CYQU2VHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMYTONZUGU2DGOJRGU. You are receiving this because you authored the thread.Message ID: @.***>
作者你好,我是了解到ABCNet后查找到最新的SwinTextSpotter项目,我觉得它应该是比ABCNet更优秀,您在SwinTextSpotter提到更新的ESTextSpotter,我在昨天尝试之后,能 执行 vis.py (需在for循环末尾添加: torch.cuda.empty_cache(),) 我的GPU都是独立的,3060,3070,4090。单张GPU资源能否支持这两个项目之一,我的资源太少了,没有8张。我即使GPU跑一张图显存也不够(在3060实验ESTextSpotter,还未测试4090,)。这两项目对资源的需求能否再下降? 这是可行的吗? 如果我对网络不管精度先剪枝确保能训练这是能被推荐的吗? 还是我这边添加设备会更好些?