orobix / fwdgrad

Implementation of "Gradients without backpropagation" paper (https://arxiv.org/abs/2202.08587) using functorch
MIT License
95 stars 7 forks source link

Test function #9

Closed belerico closed 1 year ago

belerico commented 2 years ago

We should add some tests function to test the trained model. Given some results, we should also update the readme @DavideTr8