修改模型参数中的名字

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')