Refactor cross_entropy using log_softmax and nll_loss reference…
#13
| Job | Run time |
|---|---|
| 13m 56s | |
| 13m 35s | |
| 15m 28s | |
| 16m 47s | |
| 13m 40s | |
| 13m 52s | |
| 16m 40s | |
| 1s | |
| 1h 43m 59s |