python - PyTorch:从另一个模型加载权重而不保存

Question

假设我在 PyTorch 中有两个模型，如何在不保存权重的情况下通过模型 2 的权重加载模型 1 的权重？
像这样:

model1.weights = model2.weights

在 TensorFlow 中，我可以这样做:

variables1 = model1.trainable_variables
variables2 = model2.trainable_variables
for v1, v2 in zip(variables1, variables2):
    v1.assign(v2.numpy())

Answer 1

假设你有两个相同模型的实例(必须是 nn.Module 的子类)，那么你可以使用 nn.Module.state_dict() 和 nn.Module.load_state_dict() .您可以找到州词典的简要介绍here .

model1.load_state_dict(model2.state_dict())