Thank you for the reply.
But I got exactly the same error.
I also checked the parameter ‘scale’ is in the model parameter group like below.
scale
layer1.0.weight
layer1.0.bias
layer1.1.weight
layer1.1.bias
layer2.0.weight
layer2.0.bias
layer2.1.weight
layer2.1.bias
layer3.0.weight
layer3.0.bias
layer3.1.weight
layer3.1.bias
fc1.weight
fc1.bias
fc2.weight
fc2.bias