From 2e1755c10c8831439311986ef89a7bb3f600dcf1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Do=C4=9Fan=20Patar?=
 <34914730+dpatar@users.noreply.github.com>
Date: Sat, 9 Mar 2019 18:25:01 +0300
Subject: [PATCH] cpu mode fixed

CPU mode was giving error due to missing line in the classification example.
---
 examples/classification/example.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/examples/classification/example.py b/examples/classification/example.py
index 5719f543e..4bcac6522 100755
--- a/examples/classification/example.py
+++ b/examples/classification/example.py
@@ -34,6 +34,8 @@ def get_net(caffemodel, deploy_file, use_gpu=True):
     """
     if use_gpu:
         caffe.set_mode_gpu()
+    else:
+        caffe.set_mode_cpu()
 
     # load a new model
     return caffe.Net(deploy_file, caffemodel, caffe.TEST)