DataLoader class documentation (https://github.com/fastai/fastai/blob/master/nbs/02_data.load.ipynb) says:
- device (torch.device): Defaults to default_device() which is CUDA by default. You can specify device as `torch.device(‘cpu’).
But device is not set to default_device if None (default argument value) is present on init
By exec the following:
from fastbook import *
ds = torch.rand(100,2)
dl = DataLoader(ds)
dl.one_batch().device, default_device()
I get “(device(type=‘cpu’), device(type=‘cuda’, index=0))”
Am I misunderstanding something or is a real issue?