Commit d10bbc5a authored by dzy7e's avatar dzy7e
Browse files

optimize dataset loading

parent 2ed84d49

zoo/ccip/dataset.py

+1 −1
Original line number Diff line number Diff line
@@ -69,7 +69,7 @@ class CCIPImagesDataset(ImagesDataset): 
                    _items.append((os.path.join(root, name), _maxid))

                _maxid += 1



        super(CCIPImagesDataset, self).__init__(self, _items, transform)

        super(CCIPImagesDataset, self).__init__(_items, transform)





class CharacterDataset(Dataset):