fix load_params in stnet.py

fluid/PaddleCV/video/models/model.py

@@ -241,14 +241,3 @@ def regist_model(name, model):
 def get_model(name, cfg, mode='train', args=None):
     return model_zoo.get(name, cfg, mode, args)
 
-
-if __name__ == "__main__":
-
-    class TestModel(ModelBase):
-        pass
-
-    model_zoo.regist('test', TestModel)
-    m = model_zoo.get('test', './config.txt')
-    print(m.get_train_config('batch_size'))
-    m.build_model()
-    m = model_zoo.get('test2', './config.txt')

fluid/PaddleCV/video/models/stnet/stnet.py

@@ -154,7 +154,13 @@ class STNET(ModelBase):
         return {}
 
     def load_pretrain_params(self, exe, pretrain, prog):
-        fluid.io.load_params(exe, pretrain, main_program=prog)
+	def is_parameter(var):
+             if isinstance(var, fluid.framework.Parameter):
+                 return isinstance(var, fluid.framework.Parameter) and (not ("fc_0" in var.name)) \		
+                     and (not ("batch_norm" in var.name)) and (not ("xception" in var.name)) and (not ("conv3d" in var.name))		
+
+	vars = filter(is_parameter, prog.list_vars())		
+	fluid.io.load_vars(exe, pretrain, vars=vars)
 
         param_tensor = fluid.global_scope().find_var(
             "conv1_weights").get_tensor()