param.grad[...] = proj_grad[start_idx:end_idx].view(grad_shape) # copy proj grad TypeError: 'NoneType' object does not support item assignment
param.grad[...] = proj_grad[start_idx:end_idx].view(grad_shape) # copy proj grad
TypeError: 'NoneType' object does not support item assignment