| Name |
Last commit
|
Last Update |
|---|---|---|
| .. | ||
| CrossEntropy.cpp | ||
| CrossEntropy.cu | ||
| CrossEntropy.cuh | ||
| CrossEntropy.h | ||
| LHeader.h |
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 |
|---|---|---|
| .. | ||
| CrossEntropy.cpp | 正在载入提交数据... | |
| CrossEntropy.cu | 正在载入提交数据... | |
| CrossEntropy.cuh | 正在载入提交数据... | |
| CrossEntropy.h | 正在载入提交数据... | |
| LHeader.h | 正在载入提交数据... |