Hi @albanD , if data should not be used can you please explain how to execute the following while still getting the model with the adjusted weights?
when I run:
i = 1
for w in LinModel.parameters():
    torch.nn.init.eye_(w)
    w.data = w.data*i
    i += 1

the model weights are adjusted properly by a factor of i after the loop ends. if I don’t use data then just the identity init is saved.