From cc901a6321d3becf091115163eff92fefdd7d590 Mon Sep 17 00:00:00 2001
From: JiabinYang <marsyang199376@gmail.com>
Date: Thu, 16 Aug 2018 15:31:58 +0000
Subject: [PATCH] fix representation problems

---
 03.image_classification/train.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/03.image_classification/train.py b/03.image_classification/train.py
index b323ebc..9290b6a 100644
--- a/03.image_classification/train.py
+++ b/03.image_classification/train.py
@@ -116,6 +116,7 @@ def infer(use_cuda, inference_program, params_dirname=None):
     img = load_image(cur_dir + '/image/dog.png')
 
     # inference
+    results = inferencer.infer({'pixel': img})
 
     label_list = [
         "airplane", "automobile", "bird", "cat", "deer", "dog", "frog", "horse",
-- 
GitLab