Converting Keras to Pytorch

[111191 (덕규 윤)]

Blow is my keras and torch code

Q :
I am wondering why the performance is poor when implementing with torch.

[ptrblck]

nn.CrossEntropyLoss expects raw logits, so you should remove the softmax at the end of your model.

[111191 (덕규 윤)]

Thanks you very much!