Commit d677a716 authored by Mitchell Wortsman's avatar Mitchell Wortsman
Browse files

typo

parent da524d97
...@@ -112,8 +112,6 @@ class _switchback_global_mem_efficient(torch.autograd.Function): ...@@ -112,8 +112,6 @@ class _switchback_global_mem_efficient(torch.autograd.Function):
del X del X
W_int8, state_W = quantize_global(W) W_int8, state_W = quantize_global(W)
print('in mem eff backward.')
# save for backward. # save for backward.
ctx.save_for_backward = X_int8, state_X, W_int8, state_W ctx.save_for_backward = X_int8, state_X, W_int8, state_W
......
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