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 | 正在载入提交数据... |