nan values for cudnn.SpatialCrossEntropyCriterion():cudaHalf()

14 views

Skip to first unread message

Konstantin Semyanov

unread,

Jun 16, 2017, 7:52:30 AM6/16/17

to torch7

I get all nan values while forwarding a tensor through cudnn.SpatialCrossEntropyCriterion() converted to :cudaHalf(). However, everything works fine with :cuda(). Is this the result of criterion implementation specific to the floating point precision?

Reply all

Reply to author

Forward

0 new messages