🎋 Replace all self._model.trainable_variables -> self._trainable_variables.

dathudeptrai · dathudeptrai · commit 1a64f4ed3c6e · 2020-10-14T14:09:48.000+07:00
diff --git a/tensorflow_tts/trainers/base_trainer.py b/tensorflow_tts/trainers/base_trainer.py
@@ -716,12 +716,12 @@ def _one_step_forward_per_replica(self, batch):
 
         if self._is_mixed_precision:
             scaled_gradients = tape.gradient(
-                scaled_per_replica_losses, self._model.trainable_variables
+                scaled_per_replica_losses, self._trainable_variables
             )
             gradients = self._optimizer.get_unscaled_gradients(scaled_gradients)
         else:
             gradients = tape.gradient(
-                per_replica_losses, self._model.trainable_variables
+                per_replica_losses, self._trainable_variables
             )
 
         self._optimizer.apply_gradients(zip(gradients, self._trainable_variables), 1.0)

Original file line number	Diff line number	Diff line change
`@@ -716,12 +716,12 @@ def _one_step_forward_per_replica(self, batch):`
`716`	`716`
`717`	`717`	`if self._is_mixed_precision:`
`718`	`718`	`scaled_gradients = tape.gradient(`
`719`		`- scaled_per_replica_losses, self._model.trainable_variables`
	`719`	`+ scaled_per_replica_losses, self._trainable_variables`
`720`	`720`	`)`
`721`	`721`	`gradients = self._optimizer.get_unscaled_gradients(scaled_gradients)`
`722`	`722`	`else:`
`723`	`723`	`gradients = tape.gradient(`
`724`		`- per_replica_losses, self._model.trainable_variables`
	`724`	`+ per_replica_losses, self._trainable_variables`
`725`	`725`	`)`
`726`	`726`
`727`	`727`	`self._optimizer.apply_gradients(zip(gradients, self._trainable_variables), 1.0)`