mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
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 |
||
|---|---|---|
| .. | ||
| _src | ||
| aot_autograd_optimizations.ipynb | ||
| ensembling.ipynb | ||
| jacobians_hessians.ipynb | ||
| minifier.ipynb | ||
| neural_tangent_kernels.ipynb | ||
| per_sample_grads.ipynb | ||
| whirlwind_tour.ipynb | ||