Cannot import auto_move_data decorator in Colab

[MaxTeselkin]

Hi everyone! I am trying to import auto_move_data from pytorch_lightning.core.decorators, but I get an import error:

ImportError                               Traceback (most recent call last)
[<ipython-input-3-22381b260381>](https://localhost:8080/#) in <module>()
     26 from pytorch_lightning import LightningDataModule
     27 from pytorch_lightning import LightningModule
---> 28 from pytorch_lightning.core.decorators import auto_move_data
     29 from pytorch_lightning import Trainer

ImportError: cannot import name 'auto_move_data' from 'pytorch_lightning.core.decorators' (/usr/local/lib/python3.7/dist-packages/pytorch_lightning/core/decorators.py)

Does anybody know how to solve this problem?

[rohitgr7]

it's no longer present in the package.

[MaxTeselkin]

Thanks for the answer! Is there a replacement for this decorator in the package? Or I should just put input arguments to correct device by myself (as in pure Pytorch)?

[rohitgr7]

yes. you can access self.device to move them to the device.

[MaxTeselkin]

Thanks!