From fa0ee0fb620aba7d11a39c105e882e63ba946514 Mon Sep 17 00:00:00 2001 From: SundarRajan98 Date: Fri, 22 Mar 2024 15:49:49 +0000 Subject: [PATCH] Update readme for ckpt removal instructions --- rocAL_pybind/examples/rocAL_training_example/README.md | 1 + 1 file changed, 1 insertion(+) diff --git a/rocAL_pybind/examples/rocAL_training_example/README.md b/rocAL_pybind/examples/rocAL_training_example/README.md index 40148b230..ecc5f1d8a 100644 --- a/rocAL_pybind/examples/rocAL_training_example/README.md +++ b/rocAL_pybind/examples/rocAL_training_example/README.md @@ -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/ ```