pytorch手动设置模型参数