search

shoutOutYangJie / Morph-UGATIT

a morph transfer UGATIT for image translation.
60 stars 12 forks source link

Resume problem #6

Closed vinduon closed 3 years ago

vinduon commented 3 years ago

Dear sir,

Based on your resume for ugatit, I wrote the resume for Morph-UGATIT similarly. Screenshot from 2021-04-09 17-29-51

But then I got this error: Screenshot from 2021-04-09 17-30-55

I also modified other things similar to what you did. The error is from this load_state_dict. Please help me sir. Thank you.

vinduon commented 3 years ago

BTW I fixed this issues. The problem is may from DistributedDataParallel. I just add prefix module. for current state_dict

shoutOutYangJie commented 3 years ago

BTW I fixed this issues. The problem is may from DistributedDataParallel. I just add prefix module. for current state_dict

Yes, the error is easy to correct.

Kitty-sunray commented 2 years ago

Ran into same issue. Could you please be more specific how to fix?