search

Easquel / WaveDM

Implementation code for WaveDM: Wavelet-Based Diffusion Models for Image Restoration.
https://ieeexplore.ieee.org/abstract/document/10420512
19 stars 1 forks source link

About training #3

Closed chenzhanglu closed 1 week ago

chenzhanglu commented 3 months ago

File "/public/home/acplz3jcvc/czl/WaveDM/models/arch.py", line 250, in forward x = x + enc_skip RuntimeError: The size of tensor a (74) must match the size of tensor b (75) at non-singleton dimension 2

I have tried many things and have no idea what is going on, can you reply. Thank you!

Zwette commented 2 months ago

Sorry for the late reply. You would better try inputting an image of which the size can be divided by 16.