In the markdown, you mentioned loss=trainer(data) . As a new user, I’m still figuring out how to use the models with custom data. Maybe I can add a toy_problem.py as examples and create a PR? Would really appreciate your help with this.