Fix a bug

xlnwel · Jul 25, 2019 · 151aef5 · 151aef5
1 parent 3351cd8
commit 151aef5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/basic_model/model.py b/basic_model/model.py
@@ -200,6 +200,9 @@ def __init__(self,
         super().__init__(name, args, self.graph, log_tensorboard=log_tensorboard, 
                          log_params=log_params, device=device, reuse=reuse)
 
+        # reset variable scope for model
+        self.variable_scope = self.name
+
         display_var_info(self.trainable_variables)
 
         self.model_name = args['model_name']