ddlee-cn/MuLUT

How to solve this problem?

WangXuCh opened this issue · 7 comments

Can't get attribute 'SRUnit' on <module 'common.network' from '..\common\network.py'>

I just used the model (Model_200000.pth) which you provided in the first part for the operation of the second part.

Sorry for the bug. Please replace "SRUnit" with "MuLUTUnit" in keys of the model.stated_dict().

Can't get attribute 'SRUnit' on <module 'common.network' from '..\common\network.py'>

I just used the model (Model_200000.pth) which you provided in the first part for the operation of the second part.
Excuse me, did "Model_200000.pth" appear in your python command? If yes, could you tell me where it was specifically? Thank you!

It is used as the model loading path here.

It is used as the model loading path here.

Thank you!

Please replace "SRUnit" with "MuLUTUnit" in keys of the model.stated_dict().

May I ask how you resolved it?

Please replace "SRUnit" with "MuLUTUnit" in keys of the model.stated_dict().

May I ask how you resolved it?

Please refer to this one: https://discuss.pytorch.org/t/dataparallel-changes-parameter-names-issue-with-load-state-dict/60211/2

Otherwise, you can retrain the model with step 1 code.