Fixed lipschitz_projection when different cuda device is specified

[Dom1L]

Development:

• [x] Implement feature / fix bug
• [ ] Add documentation

• [ ] Add tests

  Checks:

• [ ] Run nosetests
• [ ] Check pep8 compliance

Hey,

the lipschitz_projection was failing when a different cuda device is specified (like torch.device('cuda:1')). This fix transfers the lip_reg tensor always to the correct device.

Please test if it works for you! This should not interfere with our existing code due to the device being inferred from the weights tensor directly. So no new arguments need to be passed.

[nec4]

Great catch! Thanks for the fix. All tests pass for me. LGTM!