PSENet.pytorch
PSENet.pytorch copied to clipboard
继续训练
恢复训练时,除了在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.训练 的时候您遇到这个问题吗?