From 4525fc549e23f9cf33f3daf2cb73c86351c09cb3 Mon Sep 17 00:00:00 2001
From: wuchenghui <wuchenghui@xiaomi.com>
Date: Fri, 27 Apr 2018 16:54:35 +0800
Subject: [PATCH] fix caffe graph converter

---
 mace/python/tools/caffe_converter_lib.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/mace/python/tools/caffe_converter_lib.py b/mace/python/tools/caffe_converter_lib.py
index 9513b1fe..d0928904 100644
--- a/mace/python/tools/caffe_converter_lib.py
+++ b/mace/python/tools/caffe_converter_lib.py
@@ -423,6 +423,7 @@ class CaffeConverter(object):
         if use_winograd:
             self.convert_winograd_conv_filter_cpu(op, op_def)
         elif self.device == 'gpu':
+            self.add_tensor(weight_tensor_name, weight_data)
             buffer_type = "DW_CONV2D_FILTER" \
                 if is_depthwise else "CONV2D_FILTER"
             output_name = self.add_buffer_to_image(weight_tensor_name,
-- 
GitLab