From 0cd9a3a8ec85e8a5fdc21e5a99d4d969b9fed7f3 Mon Sep 17 00:00:00 2001
From: Takeru Miyato <miyato-t@sys.i.kyoto-u.ac.jp>
Date: Wed, 8 Jun 2016 19:18:58 +0900
Subject: [PATCH] Modification to return logit in inference_small()

---
 resnet.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/resnet.py b/resnet.py
index a056c41..24801dc 100644
--- a/resnet.py
+++ b/resnet.py
@@ -101,7 +101,10 @@ def inference_small(x,
     c['fc_units_out'] = num_classes
     c['num_blocks'] = num_blocks
     c['num_classes'] = num_classes
-    inference_small_config(x, c)
+    x = inference_small_config(x, c)
+    
+    return x
+    
 
 def inference_small_config(x, c):
     c['bottleneck'] = False