Skip to content

name 'ModelDataLoader' is not defined #2

Description

@bitdata

env: Ubuntu 16.04 x64, Python 3.6.6, Pytorch 0.2.0, web.py 0.40.dev0

NameError Traceback (most recent call last)
in ()
3 aug_tfms = [RandomRotate(20), RandomLighting(0.8, 0.8)]
4 tfms = tfms_from_model(arch, sz, aug_tfms=aug_tfms, max_zoom=1.2)
----> 5 data = ImageClassifierData.from_paths(path, tfms=tfms, test_name='test', bs=bs)
6
7 ims = np.stack([get_augs() for i in range(6)])

in from_paths(cls, path, bs, tfms, trn_name, val_name, test_name, num_workers)
60 test_fnames = read_dir(path, test_name) if test_name else None
61 datasets = cls.get_ds(FilesIndexArrayDataset, trn, val, tfms, path=path, test=test_fnames)
---> 62 return cls(path, datasets, bs, num_workers, classes=trn[2])
63
64 @classmethod

in init(self, path, datasets, bs, num_workers, classes)
6 self.get_dl(ds,shuf) for ds,shuf in [
7 (trn_ds,True),(val_ds,False),(fix_ds,False),(aug_ds,True),
----> 8 (test_ds,False),(test_aug_ds,False)
9 ]
10 ]

in (.0)
4 self.path,self.bs,self.num_workers,self.classes = path,bs,num_workers,classes
5 self.trn_dl,self.val_dl,self.fix_dl,self.aug_dl,self.test_dl,self.test_aug_dl = [
----> 6 self.get_dl(ds,shuf) for ds,shuf in [
7 (trn_ds,True),(val_ds,False),(fix_ds,False),(aug_ds,True),
8 (test_ds,False),(test_aug_ds,False)

in get_dl(self, ds, shuffle)
13 def get_dl(self, ds, shuffle):
14 if ds is None: return None
---> 15 return ModelDataLoader.create_dl(ds, batch_size=self.bs, shuffle=shuffle,
16 num_workers=self.num_workers, pin_memory=False)
17

NameError: name 'ModelDataLoader' is not defined

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions