Here I have added a notebook, that takes an image dataset and performs a classification task, leveraging Ray (Datasets and Train), Pytorch (torchvision), and the multiple GPU's available on our cluster to train and save a model that can be used for inference.
There is more work to be done here around validation, monitoring and documentation, but wanted to get the core functioning bits of this workflow reviewed and merged first. :)
This PR addresses parts of issue #4 .
Here I have added a notebook, that takes an image dataset and performs a classification task, leveraging Ray (Datasets and Train), Pytorch (torchvision), and the multiple GPU's available on our cluster to train and save a model that can be used for inference.
There is more work to be done here around validation, monitoring and documentation, but wanted to get the core functioning bits of this workflow reviewed and merged first. :)