[models.model] Implement to() method to move all computing to the designated...

[models.model] Implement to() method to move all computing to the designated accelerator device, including losses and augmentations