Ascend-Research / CascadedGaze

The official PyTorch implementation for CascadedGaze: Efficiency in Global Context Extraction for Image Restoration, TMLR'24.
MIT License
14 stars 0 forks source link

Selection of Convolutional Kernel Size and Stride in GCE Modules #2

Open shiyanshilaoshu opened 1 week ago

shiyanshilaoshu commented 1 week ago

Hello, I truly appreciate your work. However, I noticed that in the GCE module described in the paper, the stride of the convolutional layer matches the kernel size (for instance, k=[3,3,5], s=[3,3,5]), whereas in the code, it does not (k=[3,3,5], s=[2,3,4]). Could you please clarify which should be followed? 截图 2024-06-26 11-00-40

hyydrra commented 1 week ago

Hello,

Thank you for your attention to detail. Both approaches are valid and can be used based on your preference. The final outcomes are similar, but our final models were trained using the configuration in the code. In the GCE module described in the paper, we mentioned setting the stride of the convolutional layer to match the kernel size (e.g., k=[3,3,5], s=[3,3,5]). However, in the code, we used slightly smaller strides (e.g., k=[3,3,5], s=[2,3,4]) to increase the overlap between features.

shiyanshilaoshu commented 1 week ago

Thank you very much for your answer. I'm going to start the experiment now