Skip to content

Introducing dask-pytorch

Compare
Choose a tag to compare
@skirmer skirmer released this 18 Nov 17:28
· 37 commits to main since this release
d9d1645

Enables easier and smoother integration of Dask and PyTorch.

Features

  • Custom data object class S3ImageFolder allows direct-from-s3 access to files for training/inference in pytorch
  • dispatch module gives tools for distributing training tasks to cluster with minimal user fuss
  • results module gives tools for managing the results and training statistics, for better model evaluation