Update dec.py

fix bug

Author: Orange15

example/dec/dec.py

@@ -97,7 +97,7 @@ def cluster(self, X, y=None, update_interval=None):
         test_iter = mx.io.NDArrayIter({'data': X}, batch_size=batch_size, shuffle=False,
                                       last_batch_handle='pad')
         args = {k: mx.nd.array(v.asnumpy(), ctx=self.xpu) for k, v in self.args.items()}
-        z = model.extract_feature(self.feature, args, test_iter, N, self.xpu).values()[0]
+        z = model.extract_feature(self.feature, args, None, test_iter, N, self.xpu).values()[0]
         kmeans = KMeans(self.num_centers, n_init=20)
         kmeans.fit(z)
         args['dec_mu'][:] = kmeans.cluster_centers_
@@ -112,7 +112,7 @@ def ce(label, pred):
         self.y_pred = np.zeros((X.shape[0]))
         def refresh(i):
             if i%update_interval == 0:
-                z = model.extract_feature(self.feature, args, test_iter, N, self.xpu).values()[0]
+                z = model.extract_feature(self.feature, args, None, test_iter, N, self.xpu).values()[0]
                 p = np.zeros((z.shape[0], self.num_centers))
                 self.dec_op.forward([z, args['dec_mu'].asnumpy()], [p])
                 y_pred = p.argmax(axis=1)
@@ -132,7 +132,7 @@ def refresh(i):
         solver.set_iter_start_callback(refresh)
         solver.set_monitor(Monitor(50))
 
-        solver.solve(self.xpu, self.loss, args, self.args_grad,
+        solver.solve(self.xpu, self.loss, args, self.args_grad, None,
                      train_iter, 0, 1000000000, {}, False)
         self.end_args = args
         if y is not None:
@@ -153,4 +153,4 @@ def mnist_exp(xpu):
 if __name__ == '__main__':
     logging.basicConfig(level=logging.INFO)
     mnist_exp(mx.gpu(0))
-    
+