search

zhengchen1999 / DAT

PyTorch code for our ICCV 2023 paper "Dual Aggregation Transformer for Image Super-Resolution"
Apache License 2.0
350 stars 27 forks source link

评估太慢了 #39

Open zhangch-ss opened 2 months ago

zhangch-ss commented 2 months ago

发现评估validation只使用了一个batch size，速度太慢了，可以修改一下评估支持多个batch吗

zhengchen1999 commented 2 months ago

感谢建议，我们会对代码进行修改，完成后通知您。、 不过由于图片大小不一致，多batch可能不行。我们后续应该会考虑多GPU测试。

zhangch-ss commented 2 months ago

感谢建议，我们会对代码进行修改，完成后通知您。、 不过由于图片大小不一致，多batch可能不行。我们后续应该会考虑多GPU测试。

谢谢回复，请问use_chop是做的滑窗推理吗？我看代码窗口大小固定为200，我理解的对吗

zhengchen1999 commented 2 months ago

是的。 PS：感谢提醒，use_chop时可以多batch推理。