search

sql-machine-learning / elasticdl

Kubernetes-native Deep Learning Framework
https://elasticdl.org
MIT License
733 stars 113 forks source link

Create an ElasticImageFolder for PyTorch. #2486

Closed workingloong closed 3 years ago

workingloong commented 3 years ago

Users can create a dataset like:


dataset = ElasticImageFolder(
    traindir,
    transforms.Compose([
      transforms.RandomResizedCrop(224),
      transforms.RandomHorizontalFlip(),
      transforms.ToTensor(),
      normalize,
    ]))