search

Algolzw / daclip-uir

[ICLR 2024] Controlling Vision-Language Models for Universal Image Restoration. 5th place in the NTIRE 2024 Restore Any Image Model in the Wild Challenge.
https://algolzw.github.io/daclip-uir
MIT License
582 stars 30 forks source link

显卡配置要求 #51

Open jiandandan001 opened 1 month ago

jiandandan001 commented 1 month ago

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

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

Algolzw commented 1 month ago

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

jiandandan001 commented 1 month ago

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

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

Algolzw commented 1 month ago

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

jiandandan001 commented 1 month ago

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

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