%load_ext autoreload
%autoreload 2
%matplotlib inline
from exp.nb_08 import *
path = datasets.untar_data(datasets.URLs.IMAGENETTE_320)
tfms = [make_rgb, ResizeFixed(224), to_byte_tensor, to_float_tensor]
il = ImageList.from_files(path, tfms=tfms)
sd = SplitData.split_by_func(il, partial(grandparent_splitter, valid_name='val'))
ll = label_by_func(sd, parent_labeler, proc_y=CategoryProcessor())
bs=256
train_dl,valid_dl = get_dls(ll.train,ll.valid,bs, num_workers=4)
%time x,y = next(iter(train_dl))
CPU times: user 12.5 ms, sys: 169 ms, total: 182 ms Wall time: 1.99 s
%time for x,y in train_dl: x,y = x.cuda(),y.cuda()
CPU times: user 1.31 s, sys: 1.8 s, total: 3.11 s Wall time: 11.5 s