Skip to content

Commit

Permalink
[Example] Update example train maskrcnn (#2408)
Browse files Browse the repository at this point in the history
  • Loading branch information
XumingGai authored Sep 22, 2023
1 parent 3bbb535 commit fc3bc1e
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
4 changes: 2 additions & 2 deletions examples/train_maskrcnn/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ You can use `./pip_set_env.sh` to setup for GPU. It contains the following two s

```
python -m venv env_itex
source source env_itex/bin/activate
source env_itex/bin/activate
```

+ Install
Expand All @@ -48,7 +48,7 @@ pip install --upgrade pip
pip install --upgrade intel-extension-for-tensorflow[gpu]
pip install intel-optimization-for-horovod
pip install opencv-python-headless pybind11
pip install "git+https://github.com/NVIDIA/cocoapi#egg=pycocotools&subdirectory=PythonAPI"
pip install pycocotools
pip install -e "git+https://github.com/NVIDIA/dllogger#egg=dllogger"
```

Expand Down
4 changes: 2 additions & 2 deletions examples/train_maskrcnn/pip_set_env.sh
Original file line number Diff line number Diff line change
Expand Up @@ -25,5 +25,5 @@ pip install --upgrade pip
pip install --upgrade intel-extension-for-tensorflow[gpu]
pip install intel-optimization-for-horovod
pip install opencv-python-headless pybind11
pip install "git+https://github.com/NVIDIA/cocoapi#egg=pycocotools&subdirectory=PythonAPI"
pip install -e "git+https://github.com/NVIDIA/dllogger#egg=dllogger"
pip install pycocotools
pip install -e "git+https://github.com/NVIDIA/dllogger#egg=dllogger"

0 comments on commit fc3bc1e

Please sign in to comment.