search

WenmuZhou / PSENet.pytorch

A pytorch re-implementation of PSENet: Shape Robust Text Detection with Progressive Scale Expansion Network
GNU General Public License v3.0
462 stars 138 forks source link

继续训练 #97

Open Jialiang123 opened 2 years ago

Jialiang123 commented 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中 image

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

zeng-cy commented 1 year ago

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