Runtime Error from Unsqueeze

MC-E / COLA-Net-Collaborative-Attention-Network-for-Image-Restoration

Accepted by TMM 2021

41 stars 6 forks source link

Runtime Error from Unsqueeze #8

Open gauenk opened 2 years ago

gauenk commented 2 years ago

Training the model throws a runtime error. I am running pytorch version 1.11.0. I created a pull-request showing how the model needs to change for it to be differentiable. I modified the DN_Gray/ model. PR #7

MC-E commented 2 years ago

Thank you for your constructive suggestion! This modification is correct and effective!