guopengf / ReconFormer

ReconFormer: Accelerated MRI Reconstruction Using Recurrent Transformer
https://arxiv.org/abs/2201.09376
MIT License
57 stars 10 forks source link

关于显存至少得有多大可以进行模型训练 #2

Closed VincyDa closed 1 year ago

VincyDa commented 1 year ago

同学您好,有注意到你这边用的是48G的RTX8000显卡,我这边大概仅有20g的显存,想问问你那边有没有尝试过其他较小显存显卡上面跑

guopengf commented 1 year ago

把batch size变小应该可以。

VincyDa commented 1 year ago

好的,我去试试,感谢!


发件人: Pengfei Guo @.> 发送时间: 2023年2月16日 23:58 收件人: guopengf/ReconFormer @.> 抄送: VincyDa @.>; Author @.> 主题: Re: [guopengf/ReconFormer] 关于显存至少得有多大可以进行模型训练 (Issue #2)

把batch size变小应该可以。

― Reply to this email directly, view it on GitHubhttps://kor01.safelinks.protection.outlook.com/?url=https%3A%2F%2Fgithub.com%2Fguopengf%2FReconFormer%2Fissues%2F2%23issuecomment-1433316781&data=05%7C01%7C%7C819c3d0278a544b6b9a508db1036ae63%7C84df9e7fe9f640afb435aaaaaaaaaaaa%7C1%7C0%7C638121599293491623%7CUnknown%7CTWFpbGZsb3d8eyJWIjoiMC4wLjAwMDAiLCJQIjoiV2luMzIiLCJBTiI6Ik1haWwiLCJXVCI6Mn0%3D%7C3000%7C%7C%7C&sdata=Lj3AYNDEk%2B%2BVW4iPCkMQmGXCpBjTRiU6B0bDnE8y7lg%3D&reserved=0, or unsubscribehttps://kor01.safelinks.protection.outlook.com/?url=https%3A%2F%2Fgithub.com%2Fnotifications%2Funsubscribe-auth%2FARJPHZMQQAGBD4VX6F22EODWXZFDHANCNFSM6AAAAAAU54THME&data=05%7C01%7C%7C819c3d0278a544b6b9a508db1036ae63%7C84df9e7fe9f640afb435aaaaaaaaaaaa%7C1%7C0%7C638121599293491623%7CUnknown%7CTWFpbGZsb3d8eyJWIjoiMC4wLjAwMDAiLCJQIjoiV2luMzIiLCJBTiI6Ik1haWwiLCJXVCI6Mn0%3D%7C3000%7C%7C%7C&sdata=gddzfUQZQZTQ2Phazco7nPi3u0LXXNPZlX2ZuauPNVs%3D&reserved=0. You are receiving this because you authored the thread.Message ID: @.***>