marian-nmt/marian
1
1
Fork 0
mirror of https://github.com/marian-nmt/marian.git synced 2024-09-17 09:47:34 +03:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'master' of github.com:emjotde/Marian

Browse Source
This commit is contained in:
Hieu Hoang 2016-09-22 13:07:17 +01:00
commit ac1db275ab
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/mnist_benchmark.cu
View File

@ -45,6 +45,7 @@ ExpressionGraph build_graph(const std::vector<int>& dims) {
"scores");
auto cost = mean(cross_entropy(scores, y), axis=0);
//auto cost = mean(-sum(y * logsoftmax(scores), axis=1), axis=0);
auto costreg = named(
cost, "cost"
);