修改模型参数中的名字
chkpt = torch.load('weights/model_weights_StateJudgmentPre', map_location="cuda:0")
chkpt['fn_layer.weight'] = chkpt.pop('图转.weight')
chkpt['fn_layer.bias'] = chkpt.pop('图转.bias')
chkpt['evaluate.weight'] = chkpt.pop('评价.weight')
chkpt['evaluate.bias'] = chkpt.pop('评价.bias')
torch.save(chkpt, './model_weights_StateJudgmentPreEn.pth')