Skip to content

Commit

Permalink
Update readme for ckpt removal instructions
Browse files Browse the repository at this point in the history
  • Loading branch information
SundarRajan98 committed Mar 22, 2024
1 parent 2e2083f commit fa0ee0f
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions rocAL_pybind/examples/rocAL_training_example/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@ This version has been modified to use rocAL. It assumes that the dataset is raw

To run use the following command
```bash
rm *.pth.tar # Remove older checkpoints saved in the folder if the example has been run before
python3 main.py -a resnet50 --dist-url='tcp://127.0.0.1:4321' --dist-backend 'nccl' --multiprocessing-distributed --world-size 1 --rank 0 -j$(nproc) --batch-size 1024 --rocal-cpu --epochs 91 /media/imageNetCompleteDataset/
```

Expand Down

0 comments on commit fa0ee0f

Please sign in to comment.