Skip to content

Commit

Permalink
docs: resnet documentation
Browse files Browse the repository at this point in the history
added resnet documentation and script usage
  • Loading branch information
Rares Gaia committed Aug 1, 2024
1 parent 3de1642 commit 12311a0
Showing 1 changed file with 15 additions and 0 deletions.
15 changes: 15 additions & 0 deletions examples/resnet/README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
# Resnet

This script demonstrates how to run a ResNet model on multiple worlds using PyTorch. This script initializes a ResNet model on every worker. The leader sends an image to a worker for inference. After processing the image, the worker sends the predicted class back to the leader.

## Running the Script in a single host

```bash
python m8d.py --num_workers 1 --backend gloo
```

## Running the Script in multiple hosts

```bash
python m8d.py --num_workers 2 --backend nccl --multihost --addr 10.20.1.50 --rank 0
```

0 comments on commit 12311a0

Please sign in to comment.