PSENet.pytorch icon indicating copy to clipboard operation
PSENet.pytorch copied to clipboard

继续训练

Open Jialiang123 opened this issue 2 years ago • 1 comments

恢复训练时,除了在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

将load_state_dict(state_dict) 改成 model.load_state_dict(state_dict, False)

Jialiang123 avatar Mar 10 '22 11:03 Jialiang123

The kernel for PSENet/Untitled.ipynb appears to have died. It will restart automatically.训练 的时候您遇到这个问题吗?

zeng-cy avatar Nov 20 '22 00:11 zeng-cy