I have a NeuralNetwork model in which I registered in purpose two variables. When I save the state_dict to load back the model for evaluation I am getting the error:
  File "../mlchem/potentials.py", line 191, in calculate
    strict=True)
  File "/home/muammar/.local/lib/python3.7/site-packages/torch/nn/modules/module.py", line 769, in load_state_dict
    self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for NeuralNetwork:
	Unexpected key(s) in state_dict: "intercept_Cu", "slope_Cu". 

How can you load a model like this? I need those two variables to be recognized by .load_state_dict() I would appreciate any suggestions.
PS. I checked other posts but I am not having the same error as they reported.