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

mnist print to file

Browse Source
This commit is contained in:
Marcin Junczys-Dowmunt 2016-09-22 01:42:31 +02:00
parent 94bc099e6b
commit d24ee0a924
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/mnist_benchmark.cu
View File

@ -113,7 +113,10 @@ int main(int argc, char** argv) {
std::cerr << "Done." << std::endl;
ExpressionGraph g = build_graph({IMAGE_SIZE, 2048, 2048, LABEL_SIZE});
//std::cout << g.graphviz() << std::endl;
std::ofstream viz("mnist_benchmark.dot");
viz << g.graphviz() << std::endl;
viz.close();
Tensor xt({BATCH_SIZE, IMAGE_SIZE});
Tensor yt({BATCH_SIZE, LABEL_SIZE});