From 7ee583190eb275a6709470fba9c0163165fb57a2 Mon Sep 17 00:00:00 2001
From: Jin Li <engineer.tony.lee@gmail.com>
Date: Wed, 14 Aug 2019 09:18:49 +0800
Subject: [PATCH] fix bug

fix variable name
---
 lib/models.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lib/models.py b/lib/models.py
index 36ae45f..f69955f 100644
--- a/lib/models.py
+++ b/lib/models.py
@@ -48,7 +48,7 @@ def _prepare_base_model(self, base_model_name):
             self.base_model = eval(base_model_name)(pretrained=self.pretrained,
                                     feat=True, pretrained_model=base_model_dict)
         else:
-            raise ValueError('Unknown base model: {}'.format(base_model))
+            raise ValueError('Unknown base model: {}'.format(base_model_name))
 
         # classifier: (dropout) + fc
         if self.dropout == 0: