Getting MNIST training working properly with jax.nn.relu
(#98)
#310
Job | Run time |
---|---|
19s | |
48s | |
1m 7s |
jax.nn.relu
(#98)
#310
Job | Run time |
---|---|
19s | |
48s | |
1m 7s |