强化学习王者荣耀项目修改

修改模型参数中的名字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['evalua

深度学习视觉

1568人浏览 · 2022-03-29 16:38:10

深度学习视觉 · 2022-03-29 16:38:10 发布

修改模型参数中的名字


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