Quick start - load and evaluate the model

#1
by michela78 - opened

Hello, i have found your paper and huggin face model and files by looking at TRM Model informations. Would it be possible to add the code lines needed to load the model and evualate its accuracy ?
PS. The Tiny Recursive Model github is clear about data generation and pretraining but loading the model from a checkpoint and evaluation is not explicit. It could help us to have it somewhere clearly and simply explained. Best regards.

i have updated the code please check the github

Thank you for the loading details.
However using the git instructions from scratch we have modules not available. I will add an issue on the git.
Another point concern the checkpoint file : "checkpoints/cgar_50k.pth".Is it available somewhere?

Sign up or log in to comment