Open Jialiang123 opened 2 years ago
恢复训练时,除了在config.py中更改三行 pretrained = True restart_training = False checkpoint = '/home/kpl/PSENet/outputs/13.pth'
如果有如下错误 RuntimeError: Error(s) in loading state_dict for PSENet:........................................ 可尝试在dataset/data_utils.py中
将load_state_dict(state_dict) 改成 model.load_state_dict(state_dict, False)
The kernel for PSENet/Untitled.ipynb appears to have died. It will restart automatically.训练 的时候您遇到这个问题吗?
恢复训练时,除了在config.py中更改三行 pretrained = True restart_training = False checkpoint = '/home/kpl/PSENet/outputs/13.pth'
如果有如下错误 RuntimeError: Error(s) in loading state_dict for PSENet:........................................ 可尝试在dataset/data_utils.py中![image](https://user-images.githubusercontent.com/37236583/157655265-54c8f315-a590-4f13-b348-f7a93120ab64.png)
将load_state_dict(state_dict) 改成 model.load_state_dict(state_dict, False)