KerasとPyTorchの差異

Kerasを勉強した後にPyTorchを勉強して躓いたこと - Qiita
う〜ん、torch.nn.CrossEntropyLoss で数値計算の安定性のために log まで計算するのには注意・・・。Channel First と Channel Last の違いも気にしておいたほうがいいのかな？
torch.nn — PyTorch master documentationを読んでねという話。