Skip to content

Commit

Permalink
Removed blacklist
Browse files Browse the repository at this point in the history
  • Loading branch information
sean.narenthiran committed Jul 12, 2018
1 parent e7b459d commit 78e1607
Showing 1 changed file with 0 additions and 8 deletions.
8 changes: 0 additions & 8 deletions model.py
Original file line number Diff line number Diff line change
Expand Up @@ -238,14 +238,6 @@ def load_model(cls, path):
model = cls(rnn_hidden_size=package['hidden_size'], nb_layers=package['hidden_layers'],
labels=package['labels'], audio_conf=package['audio_conf'],
rnn_type=supported_rnns[package['rnn_type']], bidirectional=package.get('bidirectional', True))
# the blacklist parameters are params that were previous erroneously saved by the model
# care should be taken in future versions that if batch_norm on the first rnn is required
# that it be named something else
blacklist = ['rnns.0.batch_norm.module.weight', 'rnns.0.batch_norm.module.bias',
'rnns.0.batch_norm.module.running_mean', 'rnns.0.batch_norm.module.running_var']
for x in blacklist:
if x in package['state_dict']:
del package['state_dict'][x]
model.load_state_dict(package['state_dict'])
for x in model.rnns:
x.flatten_parameters()
Expand Down

0 comments on commit 78e1607

Please sign in to comment.