Thanks for sharing the code. But in the code, the class model does not update their parameters. Is there a mistake?