Commit Graph

31 Commits (8c19d7aa2f89a38b3a68e53c73d88af16a3de8ce)

Author SHA1 Message Date
ronnywang da10c5cf8b
[ROCM] fix softmax_with_cross_entropy_op, test=develop (#31629)
4 years ago
chajchaj 6148b87f9d
add softmax_switch for softmax_with_cross_entropy_op, test=develop (#31428)
4 years ago
Qi Li 416e47edef
[ROCM] fix softmax with loss nan in HIP platform, test=develop (#31491)
4 years ago
Qi Li db50fb6766
[ROCM] fix softmax with loss and update python scripts, test=develop (#31373)
4 years ago
Guanghua Yu d18c5e47f3
fix ignore_index check in softmax_with_cross_entropy (#31201)
4 years ago
Guanghua Yu 24ba5ee05c
merge develop conflict (#31122)
4 years ago
Zhong Hui 16fe11d71e
fix softmax cross entropy integer overflow (#30590)
4 years ago
Guo Sheng 110febdc54
Fix gradients with ignore_idx in softmax_with_cross_entropy (#28622)
4 years ago
Guanghua Yu 7779790c61
error message optimization in softmax_with_cross_entropy_op (#27772)
4 years ago
hong e362095e45
fix softmax with cross entropy out of bound; test=develop (#25549)
5 years ago
Chen Weihang 0b54d54fd8
Fix index overflow bug of the CUDA kernel loop increment (#25435)
5 years ago
tianshuo78520a d2ba91aad1
fix typo words (#22653)
5 years ago
Chen Weihang 2b941736f3 fix softmax_with_cross_entropy_fix bug, test=develop (#21810)
5 years ago
Leo Chen b3090ad406
fix synchronization problem in softmax_with_cross_entropy_op, test=develop (#21480)
5 years ago
WangXi 3c98ec90ce Fix INF bug of softmax_cross_entropy_op (#21165)
5 years ago
Zeng Jinle fff270eacd
follow comments,test=develop (#17273)
6 years ago
Kaipeng Deng a71d8fdb87
Softmax_cross_entropy op add axis (#16806)
6 years ago
Zeng Jinle ee2028a110
Add use_cuda to inplace pass (#17205)
6 years ago
Zeng Jinle 1202d3fc74
Refine model gpu memory (#16993)
6 years ago
phlrain bbfc82cc42 softmax corss entropy support high rank
6 years ago
chengduo efca4de78e Fix cross_entropy bug (#16236)
6 years ago
Wu Yi fd85418329
[Feature] support mix precision training for resnet (#14899)
6 years ago
Yu Yang 7604b1ad51 Fix Eigen macro when using GPU
6 years ago
sneaxiy 5e5d2223a1 test=develop
6 years ago
Bai Yifan e69d9c845b code fix (#13365)
7 years ago
Bai Yifan faf8ad2436
Add ignore_index in cross_entropy op (#13217)
7 years ago
sneaxiy 1b4515f6db refine softmax_with_cross_entropy
7 years ago
qingqing01 b5a16dca20
Fix a critical bug in softmax_with_cross_entropy_op backward. (#9120)
7 years ago
qingqing01 24509f4af9 Fix the grammar in copyright. (#8403)
7 years ago
Yi Wang fc374821dd Correct #include path
7 years ago
Yi Wang 90648f336d Move file to fluid/; Edit CMakeLists.txt
7 years ago