♻ Freeze -> train.

dathudeptrai · dathudeptrai · commit 5dd78f40dcbc · 2020-10-15T12:48:39.000+07:00
diff --git a/tensorflow_tts/trainers/base_trainer.py b/tensorflow_tts/trainers/base_trainer.py
@@ -647,11 +647,11 @@ def compile(self, model, optimizer):
 
     def _train_vars(self):
         if self.config["var_train_expr"]:
-            list_freeze_var = self.config["var_train_expr"].split("|")
+            list_train_var = self.config["var_train_expr"].split("|")
             return [
                 v
                 for v in self._model.trainable_variables
-                if self._check_string_exist(list_freeze_var, v.name)
+                if self._check_string_exist(list_train_var, v.name)
             ]
         return self._model.trainable_variables
 

Original file line number	Diff line number	Diff line change
`@@ -647,11 +647,11 @@ def compile(self, model, optimizer):`
`647`	`647`
`648`	`648`	`def _train_vars(self):`
`649`	`649`	`if self.config["var_train_expr"]:`
`650`		`- list_freeze_var = self.config["var_train_expr"].split("\|")`
	`650`	`+ list_train_var = self.config["var_train_expr"].split("\|")`
`651`	`651`	`return [`
`652`	`652`	`v`
`653`	`653`	`for v in self._model.trainable_variables`
`654`		`- if self._check_string_exist(list_freeze_var, v.name)`
	`654`	`+ if self._check_string_exist(list_train_var, v.name)`
`655`	`655`	`]`
`656`	`656`	`return self._model.trainable_variables`
`657`	`657`