Skip to content

Commit

Permalink
Update shard-eg readme (#2005)
Browse files Browse the repository at this point in the history
  • Loading branch information
wangkl2 authored Sep 18, 2023
1 parent a916104 commit 023b4fa
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions examples/train_resnet50_with_autoshard/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@ Modify `/path/to/tf-models` accordingly, here `~/tf-models` as an example.
```bash
cd official/legacy/image_classification/resnet/
mkdir output
export PYTHONPATH=$PYTHONPATH:/path/to/tf-models:$PWD]
export PYTHONPATH=$PYTHONPATH:/path/to/tf-models:$PWD
```

## Executes the Example with Python API
Expand Down Expand Up @@ -128,7 +128,7 @@ python resnet_ctl_imagenet_main.py \
```

The following output log indicates XPUAutoShard has been enabled successfully:<br>
`I itex/core/graph/tfg_optimizer_hook/tfg_optimizer_hook.cc:289] Run AutoShard pass successfully`
`I itex/core/graph/tfg_optimizer_hook/tfg_optimizer_hook.cc:280] Run AutoShard pass successfully`

## Example Output
With successful execution, it will print out the following results:
Expand Down

0 comments on commit 023b4fa

Please sign in to comment.