mirror of
https://github.com/malarinv/tacotron2
synced 2026-03-08 01:32:35 +00:00
hparams.py: adding ignore_layers argument to ignore text embedding layers when warm_starting
This commit is contained in:
@@ -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 #
|
||||||
|
|||||||
Reference in New Issue
Block a user