Skip to content

Commit

Permalink
encoder freeze var typo
Browse files Browse the repository at this point in the history
fix var_list argument typo
  • Loading branch information
Rayhane-mamah committed Jan 5, 2019
1 parent 4dc16ce commit 0ae2901
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tacotron/models/tacotron.py
Expand Up @@ -377,7 +377,7 @@ def add_optimizer(self, global_step):
with tf.device(tf.train.replica_device_setter(ps_tasks=1, ps_device="/cpu:0", worker_device=gpus[i])):
with tf.variable_scope('optimizer') as scope:
update_vars = [v for v in self.all_vars if not ('inputs_embedding' in v or 'encoder_' in v)] if hp.tacotron_fine_tuning else None
gradients = optimizer.compute_gradients(self.tower_loss[i], list_var=update_vars)
gradients = optimizer.compute_gradients(self.tower_loss[i], var_list=update_vars)
tower_gradients.append(gradients)

# 3. Average Gradient
Expand Down

0 comments on commit 0ae2901

Please sign in to comment.