I’m not completely sure if I understood you correctly. It is better to move the buffer to the GPU, right? But how can I do that?
Can I just do something like
self.v_old = [torch.zeros_like(p.data).to(device) for p in model.parameters()]

or is this not possible?