AECR-Net icon indicating copy to clipboard operation
AECR-Net copied to clipboard

Published 20 hours ago • verified publisher icon GlassyWu

Resume released trained model

Open PaulTHong opened this issue 2 years ago • 3 comments

When I resume the released model, there is an error: "Unexpected key(s) in state_dict: "mix4.w", "mix5.w"". mix4 and mix5 don't appear in the network constructure. So the released model is not matched with the released code. Am I right? Thanks.

PaulTHong avatar Mar 06 '22 12:03 PaulTHong

have you figure out ? the net code has 'mix1' and 'mix2', and if i just rename the mix4 to mix1, mix5 to mix2, the psnr result seems well

wangFanFanFan avatar Mar 09 '22 09:03 wangFanFanFan

Oh, thanks! I will have a try. Have you tried to train from scratch? I can't reach the evaluation metrics as shown in the paper.

PaulTHong avatar Mar 11 '22 03:03 PaulTHong

please， is the predict code named funtion test in train_aecrnet.py? How to revise it to apply a picture of JPG? Thanks!

Bruce-WangGF avatar Mar 22 '22 07:03 Bruce-WangGF