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

debug

Browse Source
This commit is contained in:
Hieu Hoang 2016-09-14 13:42:59 +01:00
parent 9643f52aa5
commit 04a620db19
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/test.cu
View File

@ -24,7 +24,7 @@ int main(int argc, char** argv) {
Expr scores = dot(x, w) + b;
Expr lr = softmax(scores, axis=1, name="pred");
Expr graph = -mean(sum(y * log(lr), axis=1), axis=0, name="cost");
cerr << "lr=" << Debug(lr.val().shape()) << endl;
//cerr << "lr=" << Debug(lr.val().shape()) << endl;
int numofdata;
vector<float> images = datasets::mnist::ReadImages("../examples/mnist/t10k-images-idx3-ubyte", numofdata, IMAGE_SIZE);
@ -50,7 +50,8 @@ int main(int argc, char** argv) {
std::cerr << "lr: " << Debug(lr.val().shape()) << endl;
std::cerr << "Log-likelihood: " << Debug(graph.val().shape()) << endl ;
std::cerr << "scores=" << scores.val().Debug() << endl;
//std::cerr << "scores=" << scores.val().Debug() << endl;
std::cerr << "lr=" << lr.val().Debug() << endl;
graph.backward();