This commit is contained in:
Mitchell Wortsman 2023-04-08 19:36:17 +00:00
parent da524d97c9
commit d677a71607

View File

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