epoch          1        10        50       200      1000      4000     final test acc
------------------------------------------------------------------------------------
logits in    20.3%     75.9%     85.6%     85.9%     85.0%     82.5%       82.5%
softmax in   12.6%     24.7%     62.4%     85.6%     86.4%     85.4%       85.4%