SofaDefrost / SofaGym

Machine Learning framework for Sofa
68 stars 10 forks source link

Fix model loading and evaluation #45

Closed samuelmyoussef closed 8 months ago

samuelmyoussef commented 10 months ago

Fixed an issue where the model loading was not done correctly because the training and the testing environment weren't wrapped and normalized in the same way, causing the evaluation of the loaded model to differ from the resulting training.

Changes:

samuelmyoussef commented 9 months ago

Ok, will try to do this in future PRs. I already made a PR to update some stuff in the README, I could make another commit to it to reflect the changes from this PR.