[Tools:Bugfix] Fix wrong path of schema head file for quantization tools

Zhuoyao1012 · Oct 30, 2019 · 9ad653d · 9ad653d
1 parent 908f4b9
commit 9ad653d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/tools/quantization/Helper.hpp b/tools/quantization/Helper.hpp
@@ -10,7 +10,7 @@
 #include <string>
 #include "ImageProcess.hpp"
 #include "Tensor.hpp"
-#include "converter/source/IR/MNN_generated.h"
+#include "MNN_generated.h"
 #include "logkit.h"
 
 #pragma once

diff --git a/tools/quantization/calibration.hpp b/tools/quantization/calibration.hpp
@@ -14,7 +14,7 @@
 #include "ImageProcess.hpp"
 #include "Interpreter.hpp"
 #include "TensorStatistic.hpp"
-#include "converter/source/IR/MNN_generated.h"
+#include "MNN_generated.h"
 
 // Calibration find the optimal threshold according to KL-divergence
 // process: the below process is applied on the whole Conv|DepthwiseConv layers

diff --git a/tools/train/CMakeLists.txt b/tools/train/CMakeLists.txt
@@ -21,12 +21,11 @@ include_directories(${MNN_PATH}/express/include)
 include_directories(source/transform)
 include_directories(source/train)
 file(GLOB BASIC_INCLUDE ${MNN_PATH}/include/*)
-file(GLOB SCHEMA ${MNN_PATH}/tools/converter/source/IR/*.h ${MNN_PATH}/tools/converter/source/IR/flatbuffers/*.h )
 file(GLOB TRANSFORMER ./source/transform/*)
 file(GLOB OPERATOR ./source/operator/*)
 
 include_directories(source/operator)
-add_library(transformer SHARED ${TRANSFORMER} ${SCHEMA} ${BASIC_INCLUDE})
+add_library(transformer SHARED ${TRANSFORMER} ${BASIC_INCLUDE})
 target_link_libraries(transformer MNN MNN_Express)
 
 add_executable(transformer.out ./source/exec/transformer.cpp)