hacky

uTensor · Mar 30, 2018 · 3ea8b1f · 3ea8b1f
1 parent f9063b4
commit 3ea8b1f
Show file tree

Hide file tree

Showing 4 changed files with 175 additions and 171 deletions.
diff --git a/image.h b/image.h
@@ -2,7 +2,7 @@
 #define IMAGE_H
 
 #include <vector>
-#include "utensor/tensor.hpp"
+#include "uTensor/core/tensor.hpp"
 
 template<typename T, template<typename> typename TENSOR=RamTensor>
 class Image {

diff --git a/main.cpp b/main.cpp
@@ -123,10 +123,10 @@ int main()
             img28_2.get_data()->resize({1, 784});
             pc.printf("Creating Graph\n\r");
 
-            get_deep_mlp_ctx(ctx, img28_2.get_data());
+            get_quant_model_ctx(ctx, img28_2.get_data());
             pc.printf("Evaluating\n\r");
             ctx.eval();
-            S_TENSOR prediction = ctx.get({"y_pred:0"});
+            S_TENSOR prediction = ctx.get({"Prediction/y_pred:0"});
             int result = *(prediction->read<int>(0,0));
 
             pc.printf("Number guessed %d\n\r", result);