pytorch/functorch/notebooks
Shayekh Bin Islam e8005781be Softmax in functorch example fixed (#107988)
The output of softmax was overwritten by the output of fc2 in the following line. So, the output of the softmax is never utilized. Now, the final output of the model includes softmax.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/107988
Approved by: https://github.com/zou3519
2023-09-05 17:18:48 +00:00
..
_src Apply UFMT to all non test/torch files (#106205) 2023-07-29 02:56:24 +00:00
aot_autograd_optimizations.ipynb Fix some typos, mostly "that that" (#106901) 2023-08-10 19:46:53 +00:00
ensembling.ipynb [BE] [2/3] Rewrite super() calls in functorch and torch (#94588) 2023-02-10 21:16:33 +00:00
jacobians_hessians.ipynb
minifier.ipynb
neural_tangent_kernels.ipynb [BE] [2/3] Rewrite super() calls in functorch and torch (#94588) 2023-02-10 21:16:33 +00:00
per_sample_grads.ipynb Softmax in functorch example fixed (#107988) 2023-09-05 17:18:48 +00:00
whirlwind_tour.ipynb