Torch Device

TIL that the integer is just the accelerator, i.e. these are all the same:

model.to(1)
model.to("cuda:1")
model.to(torch.device("cuda:1"))

I found this in the docs: https://docs.pytorch.org/docs/main/tensor_attributes.html