bug fix

wxhucong · Oct 10, 2016 · f74c0b5 · f74c0b5
1 parent be162ba
commit f74c0b5
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 5 deletions.
diff --git a/.gitignore b/.gitignore
@@ -3,4 +3,6 @@ result
 data
 *.pyc
 log.txt
+*.json
+*.params
 .idea
diff --git a/center_loss.py b/center_loss.py
@@ -26,9 +26,9 @@ def update(self, labels, preds):
 
 
 # define some metric of center_loss
-class CenterLoss(mx.metric.EvalMetric):
+class CenterLossMetric(mx.metric.EvalMetric):
     def __init__(self):
-        super(CenterLoss, self).__init__('center_loss')
+        super(CenterLossMetric, self).__init__('center_loss')
 
     def update(self, labels, preds):
         self.sum_metric = + preds[1].asnumpy()[0]

diff --git a/data.py b/data.py
@@ -5,7 +5,7 @@
 
 # code to automatically download dataset
 mxnet_root = ''
-sys.path.append(os.path.join( mxnet_root, '/tests/python/common'))
+sys.path.append(os.path.join( mxnet_root, 'tests/python/common'))
 import get_data
 import mxnet as mx
 

diff --git a/train_model.py b/train_model.py
@@ -141,8 +141,8 @@ def fit(args, network, data_loader, data_shape, batch_end_callback=None, pattern
 
     # custom metric
     eval_metrics = mx.metric.CompositeEvalMetric()
-    eval_metrics.add(Accuracy)
-    eval_metrics.add(CenterLoss())
+    eval_metrics.add(Accuracy())
+    eval_metrics.add(CenterLossMetric())
 
     model.fit(
         X                  = train,
-Original file line number
+Diff line change
@@ Expand Up / @@ -3,4 +3,6 @@ result @@
     data
     *.pyc
     log.txt
+    *.json
+    *.params
     .idea