ImportError: cannot import name 'PY3' from 'torch._six'
I tried running the example from the section 'RAPID PROTOTYPING TEMPLATES' from the pytorch lightning documentation (https://pytorch-lightning.readthedocs.io/en/latest/starter/rapid_prototyping_templates.html) but gives the following error: "ImportError: cannot import name 'PY3' from 'torch._six' (/usr/local/lib/python3.7/dist-packages/torch/_six.py)
" when trying to make this import from pl_bolts.datasets import DummyDataset. Is it a probem of versions?