@ptrblck I tried with None but that did not work. However, if I set those custom variables to zero the model is loaded with strict=True. This is the commit 233aaa6 11 that fixed it.
Now, there is something weird… When using the state dictionary that is loaded to predict the points in the training set and verify things, then they don’t match… I will investigate more.