That sounds right - and change learn.crit
to use cross_entropy. Let us know if you get it working.
1 Like
That sounds right - and change learn.crit
to use cross_entropy. Let us know if you get it working.