diff --git a/README.md b/README.md index c0761ad8..6bee418b 100644 --- a/README.md +++ b/README.md @@ -24,6 +24,7 @@ Exporting some paths for CuDNN may be required (put it, for example, in your `.b Compilation with `cmake > 3.5`: mkdir build + cd build cmake .. make -j diff --git a/src/test.cu b/src/test.cu index 56156fee..a7b5d874 100644 --- a/src/test.cu +++ b/src/test.cu @@ -6,14 +6,14 @@ int main(int argc, char** argv) { using namespace marian; using namespace keywords; - auto x = input(shape={whatevs, 784}, name="X"); - auto y = input(shape={whatevs, 10}, name="Y"); + Expr x = input(shape={whatevs, 784}, name="X"); + Expr y = input(shape={whatevs, 10}, name="Y"); - auto w = param(shape={784, 10}, name="W0"); - auto b = param(shape={1, 10}, name="b0"); + Expr w = param(shape={784, 10}, name="W0"); + Expr b = param(shape={1, 10}, name="b0"); - auto lr = softmax(dot(x, w) + b, axis=1, name="pred"); - auto graph = -mean(sum(y * log(lr), axis=1), axis=0, name="cost"); + Expr lr = softmax(dot(x, w) + b, axis=1, name="pred"); + Expr graph = -mean(sum(y * log(lr), axis=1), axis=0, name="cost"); Tensor tx({500, 784}, 1); Tensor ty({500, 10}, 1); @@ -46,4 +46,4 @@ int main(int argc, char** argv) { //opt.run(); return 0; -} \ No newline at end of file +}