From bb6761349354ee914909a42208e4820929612069 Mon Sep 17 00:00:00 2001
From: rafaelvalle <rafaelvalle@nvidia.com>
Date: Fri, 15 Mar 2019 17:28:50 -0700
Subject: [PATCH] hparams.py: adding ignore_layers argument to ignore text
 embedding layers when warm_starting

---
 hparams.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/hparams.py b/hparams.py
index 55f39e6..8886f18 100644
--- 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             #