Skip to content

GitLab

Unverified Commit 06d035e3 authored by Carl Case's avatar Carl Case Committed by GitHub
Browse files

Merge pull request #13 from cclauss/patch-1 

Type in variable name: hidden --> hiddens (w/ trailing 's')
parents 523be483 2e32e245
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
apex/amp/wrap.py
View file @ 06d035e3
...@@ -191,11 +191,11 @@ def rnn_cast(backend, fn, verbose=False): ...@@ -191,11 +191,11 @@ def rnn_cast(backend, fn, verbose=False):
# 2) Inputs: either a tuple (for LSTM) or single tensor # 2) Inputs: either a tuple (for LSTM) or single tensor
if isinstance(hiddens, tuple): if isinstance(hiddens, tuple):
new_args.append(tuple(cast_fn(x) for x in hiddens)) new_args.append(tuple(cast_fn(x) for x in hiddens))
elif utils.is_fp_tensor(hidden): elif utils.is_fp_tensor(hiddens):
new_args.append(cast_fn(hidden)) new_args.append(cast_fn(hiddens))
else: else:
# Hidden can, in principle, be `None` -- pass through # Hiddens can, in principle, be `None` -- pass through
new_args.append(hidden) new_args.append(hiddens)
# 3) Batch sizes (0.4 or later only) # 3) Batch sizes (0.4 or later only)
if len(fargs) == 4: if len(fargs) == 4:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment