mirror of
https://github.com/marian-nmt/marian.git
synced 2024-09-19 02:37:14 +03:00
Merge branch 'master' of github.com:emjotde/Marian
This commit is contained in:
commit
ac1db275ab
@ -45,6 +45,7 @@ ExpressionGraph build_graph(const std::vector<int>& dims) {
|
|||||||
"scores");
|
"scores");
|
||||||
|
|
||||||
auto cost = mean(cross_entropy(scores, y), axis=0);
|
auto cost = mean(cross_entropy(scores, y), axis=0);
|
||||||
|
//auto cost = mean(-sum(y * logsoftmax(scores), axis=1), axis=0);
|
||||||
auto costreg = named(
|
auto costreg = named(
|
||||||
cost, "cost"
|
cost, "cost"
|
||||||
);
|
);
|
||||||
|
Loading…
Reference in New Issue
Block a user