You can list and verify that the parameters m1, m2 and m3 are present in your model as follow:
for name, param in net.named_parameters():
print(name, type(param.data), param.size())
The code is from this discussion:
Cheers !
You can list and verify that the parameters m1, m2 and m3 are present in your model as follow:
for name, param in net.named_parameters():
print(name, type(param.data), param.size())
The code is from this discussion:
Cheers !