torch_model_state
