COLA-Net-Collaborative-Attention-Network-for-Image-Restoration icon indicating copy to clipboard operation
COLA-Net-Collaborative-Attention-Network-for-Image-Restoration copied to clipboard

Published 20 hours ago • verified publisher icon MC-E

Runtime Error from Unsqueeze

Open gauenk opened this issue 2 years ago • 1 comments

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

gauenk avatar Jun 30 '22 14:06 gauenk

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

MC-E avatar Sep 28 '22 07:09 MC-E