Merge pull request BVLC#18 from lmthang/dev

Add param_set to apollocaffe.update

python/apollocaffe/cpp/_apollocaffe.pyx

@@ -320,14 +320,16 @@ cdef class ApolloNet:
     def backward(self):
         for layer_name in self.active_layer_names()[::-1]:
             self.backward_layer(layer_name)
-    def update(self, lr, momentum=0., clip_gradients=-1, weight_decay=0.):
+    def update(self, lr, momentum=0., clip_gradients=-1, weight_decay=0., param_set=None):
         diffnorm = self.diff_l2_norm() 
         clip_scale = 1.
         if clip_gradients > 0:
             if diffnorm > clip_gradients:
                 clip_scale = clip_gradients / diffnorm
         params = self.params
-        for param_name in self.active_param_names():
+        if param_set is None:
+            param_set = self.active_param_names()
+        for param_name in param_set:
             self.update_param(params[param_name],
                               lr * clip_scale * self.param_lr_mults(param_name),
                               momentum,