Fix calls to .item()

• Replace call to item by detach (see https://pytorch-lightning.readthedocs.io/en/latest/guides/speed.html#things-to-avoid for details)