Skip to content

GitLab

Commit b6d4369b authored by Lawrence McAfee's avatar Lawrence McAfee
Browse files

jan 7 beta

parent ed0c8714
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 5 deletions
megatron/schedules.py
View file @ b6d4369b
...@@ -632,11 +632,11 @@ def forward_backward_pipelining_without_interleaving(forward_step_func, data_ite ...@@ -632,11 +632,11 @@ def forward_backward_pipelining_without_interleaving(forward_step_func, data_ite
if not forward_only: if not forward_only:
# >>> # >>>
if input_tensor[0] is not None: # if input_tensor[0] is not None:
from lutil import pax # from lutil import pax
pax({ # pax({
"input_tensor" : input_tensor, # "input_tensor" : input_tensor,
}) # })
# <<< # <<<
assert_viewless_tensor(input_tensor) assert_viewless_tensor(input_tensor)
assert_viewless_tensor(output_tensor) assert_viewless_tensor(output_tensor)
......
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