如何将层中的某些权重设置为特定的数字

Question

我想问一下，是否可以将层中的一些权重设置为预训练模型的特定数量或零。

例如，我想下载乐存模型，并将最后一层的一些权重设置为e.x 4，并计算精度。

我该怎么做？

Answer 1

您正在接受"净手术"。在python中，您可以加载网络并直接访问存储的权重。然后你可以随心所欲地调整它们，并保存修改后的网络。