PyTorch Forums
Categorical cross entropy loss function equivalent in PyTorch
ahmadreza9
(Ahmadreza)
June 12, 2020, 11:01am
3
@ptrblck
, I want something like below image.
IMG_20200612_153021
1080×1478 154 KB
2020-06-12_19h22_35
611×653 42.9 KB
1 Like
Can't pass LongTensor to custom model (expected scalar type Long but found Float)
show post in topic