1. update Spread, ReduceSumAll, Multiply, Unsqueeze, Softmax, CrossEntropy with float16 datatype 2. modify the implementation of fnnlm to support float16 computation, but it remains some bugs and the loss is nan
Name |
Last commit
|
Last Update |
---|---|---|
.. | ||
Dropout.cpp | 正在载入提交数据... | |
Dropout.cu | 正在载入提交数据... | |
Dropout.cuh | 正在载入提交数据... | |
Dropout.h | 正在载入提交数据... | |
DropoutWithIndex.cpp | 正在载入提交数据... | |
DropoutWithIndex.cu | 正在载入提交数据... | |
DropoutWithIndex.cuh | 正在载入提交数据... | |
DropoutWithIndex.h | 正在载入提交数据... | |
FHeader.h | 正在载入提交数据... | |
HardTanH.cpp | 正在载入提交数据... | |
HardTanH.cu | 正在载入提交数据... | |
HardTanH.cuh | 正在载入提交数据... | |
HardTanH.h | 正在载入提交数据... | |
Identity.cpp | 正在载入提交数据... | |
Identity.h | 正在载入提交数据... | |
LogSoftmax.cpp | 正在载入提交数据... | |
LogSoftmax.cu | 正在载入提交数据... | |
LogSoftmax.cuh | 正在载入提交数据... | |
LogSoftmax.h | 正在载入提交数据... | |
Loss.cpp | 正在载入提交数据... | |
Loss.cu | 正在载入提交数据... | |
Loss.cuh | 正在载入提交数据... | |
Loss.h | 正在载入提交数据... | |
Rectify.cpp | 正在载入提交数据... | |
Rectify.cu | 正在载入提交数据... | |
Rectify.cuh | 正在载入提交数据... | |
Rectify.h | 正在载入提交数据... | |
Sigmoid.cpp | 正在载入提交数据... | |
Sigmoid.cu | 正在载入提交数据... | |
Sigmoid.cuh | 正在载入提交数据... | |
Sigmoid.h | 正在载入提交数据... | |
Softmax.cpp | 正在载入提交数据... | |
Softmax.cu | 正在载入提交数据... | |
Softmax.cuh | 正在载入提交数据... | |
Softmax.h | 正在载入提交数据... |