Skip to content
Snippets Groups Projects
Commit cc8ad9a9 authored by Fahad Khalid's avatar Fahad Khalid
Browse files

Added readme for the PyTorch directory.

parent be3dce52
No related branches found
No related tags found
1 merge request!4Updated to use Tensorflow2
# Notes
The source code sample was taken from the Horovod examples repository
[here](https://github.com/horovod/horovod/tree/master/examples/pytorch)
(last checked: April 27, 2021). The sample has been slightly modified. Our
changes are limited to,
* The data loading mechanism.
* Removal of `filelock` to eliminate dependence on a package that is not
available on the supercomputers.
* A few additional comments pertaining to our custom data loading mechanism.
**Note:** All newly added statements follow a comment beginning with `[HPCNS]`.
All statements that demonstrate the use of Horovod follow a comment beginning
with `[Horovod]` (as added by Horovod developers).
The following sample is included:
1. `mnist.py`: A simple training program for an MNIST classifier that
uses Horovod for data distribution.
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment