Browse Source

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

master
rafaelvalle 5 years ago
parent
commit
bb67613493
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      hparams.py

+ 1
- 0
hparams.py View File

@ -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…
Cancel
Save