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!
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.