hparams.py: adding ignore_layers argument to ignore text embedding layers when warm_starting

2019-03-15 17:28:50 -07:00 · 2019-03-15 17:28:50 -07:00 · bb67613493
parent af1f71a975
commit bb67613493
1 changed files with 1 additions and 0 deletions
--- a/hparams.py
+++ b/hparams.py
@ -19,6 +19,7 @@ def create_hparams(hparams_string=None, verbose=False):
        dist_url="tcp://localhost:54321",
        cudnn_enabled=True,
        cudnn_benchmark=False,
+        ignore_layers=['embedding.weight'],

        ################################
        # Data Parameters             #