CUDA of memory during testing

[Grassyue]

It’s an awesome work, and thank you for sharing.😄

When I apply DA-CLIP to train on my own dataset for a dehazing task, I encounter an error during testing: "CUDA out of memory." It seems like the issue is due to the large size of my test images (1024x1024). Could you please provide some suggestions to fix this? Thank you very much!