zhuyr97 / WGWS-Net

77 stars 4 forks source link

RuntimeError: Error(s) in loading state_dict for UNet #11

Open 1001kathy opened 10 months ago

1001kathy commented 10 months ago

A RuntimeError occurs when selecting the pre-trained model of Setting1 to train Setting2, indicating that there are missing keys in the model's state_dict. I tried to use the seting1 pre-trained model you provided and this error also occurred. There was a problem loading the pre-trained model into the UNet network. I wonder why this error occured. Screenshot from 2024-01-19 16-32-56