Medical-SAM-Adapter icon indicating copy to clipboard operation
Medical-SAM-Adapter copied to clipboard

显存问题

Open xiaohongzaizhe opened this issue 2 years ago • 6 comments

我使用了48G的显存的显卡,parser.add_argument('-chunk', type=int, default=5 , help='crop volume depth') #原本96 parser.add_argument('-num_sample', type=int, default=1 , help='sample pos and neg') parser.add_argument('-roi_size', type=int, default=96 , help='resolution of roi') parser.add_argument('-b', type=int, default=1, help='batch size for dataloader')都会爆显存,而太低的参数会导致loss=1 模式崩溃

xiaohongzaizhe avatar Jul 19 '23 05:07 xiaohongzaizhe

Bot detected the issue body's language is not English, translate it automatically. 👯👭🏻🧑‍🤝‍🧑👫🧑🏿‍🤝‍🧑🏻👩🏾‍🤝‍👨🏿👬🏿


Title: Video memory problem

Issues-translate-bot avatar Jul 19 '23 05:07 Issues-translate-bot

请问原作者是在单卡显存多少的情况下训练的?

xiaohongzaizhe avatar Jul 19 '23 05:07 xiaohongzaizhe

Bot detected the issue body's language is not English, translate it automatically. 👯👭🏻🧑‍🤝‍🧑👫🧑🏿‍🤝‍🧑🏻👩🏾‍🤝‍👨🏿👬🏿


May I ask how the original author trained with the amount of video memory on a single card?

Issues-translate-bot avatar Jul 19 '23 05:07 Issues-translate-bot

Bot detected the issue body's language is not English, translate it automatically. 👯👭🏻🧑‍🤝‍🧑👫🧑🏿‍🤝‍🧑🏻👩🏾‍🤝‍👨🏿👬🏿


How can I support Doka training?

Issues-translate-bot avatar Jul 19 '23 06:07 Issues-translate-bot

try smaller batch size or image resolution

WuJunde avatar Jan 07 '24 15:01 WuJunde

Can 3090 be trained

2308700388 avatar Feb 20 '24 10:02 2308700388