显卡配置要求

[jiandandan001]

感谢分享这个优秀工作的代码。

请问要训练和测试这个模型，对于显卡最低要求要达到什么层次，在3090这种显卡上能运行吗？谢谢！

[Algolzw]

你好，如果只测试的话一个3090应该可以(但输入尺寸不能太大)。训练则需要使用很小的patch size和batch size才行，我目前只在A100上试过（最小40GB显存）。你也可以尝试加一个downsample操作例如这里。

[jiandandan001]

你好，如果只测试的话一个3090应该可以(但输入尺寸不能太大)。训练则需要使用很小的patch size和batch size才行，我目前只在A100上试过（最小40GB显存）。你也可以尝试加一个downsample操作例如这里。

谢谢您的回复和解答。请问您是说在40GB的A100上做过训练吗？

[Algolzw]

是的，不过40GB也需要设置小一点的参数。最好也加上downsample操作。

[jiandandan001]

是的，不过40GB也需要设置小一点的参数。最好也加上downsample操作。

明白，谢谢您，我去尝试下。