请教一些算力方面的问题

[whx57]

请问这个做全球的预报，大概需要什么样的算力？以及用作者你的算力训练的时间开销多大呢。因为我尝试了很多次感觉两张3090跑不起来qvq

[taohan10200]

我复现的方法里面，有的模型显存开销很大，对算力要求比较大，但是有显存占用少的方法两张3090也可以，具体可参考附录的计算复杂度对比。另外在算力有限的情况下可以减小batchsize。

[whx57]

哈哈，感谢

…

------------------ 原始邮件 ------------------ 发件人: "taohan10200/WEATHER-5K" ***@***.***>; 发送时间: 2024年7月30日(星期二) 晚上11:34 ***@***.***>; ***@***.******@***.***>; 主题: Re: [taohan10200/WEATHER-5K] 请教一些算力方面的问题 (Issue #2) 我复现的方法里面，有的模型显存开销很大，对算力要求比较大，但是有显存占用少的方法两张3090也可以，具体可参考附录的计算复杂度对比。另外在算力有限的情况下可以减小batchsize。 — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: ***@***.***>