mirror of https://github.com/malarinv/tacotron2
hparams.py: adding ignore_layers argument to ignore text embedding layers when warm_starting
parent
af1f71a975
commit
bb67613493
|
|
@ -19,6 +19,7 @@ def create_hparams(hparams_string=None, verbose=False):
|
||||||
dist_url="tcp://localhost:54321",
|
dist_url="tcp://localhost:54321",
|
||||||
cudnn_enabled=True,
|
cudnn_enabled=True,
|
||||||
cudnn_benchmark=False,
|
cudnn_benchmark=False,
|
||||||
|
ignore_layers=['embedding.weight'],
|
||||||
|
|
||||||
################################
|
################################
|
||||||
# Data Parameters #
|
# Data Parameters #
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue