Skip to content

Commit

Permalink
Adds ipynb for BERT classifier fine tuning with IMDB using launch_ben…
Browse files Browse the repository at this point in the history
…chmark.py (#135)

* Add notebook with BERT classifier fine tuning using IMDB

* Doc updates

* Update volume mounts in the docker command

* Clearing output from the notebook

* Clearing output, adding newline

* Grabbing benchmark_data_types_perf_comparison.ipynb from develop

* Update README to update to the detached container suggested by Ebi

* Add docker logs command and formatting docker run command

* Update gitignore to add .ipynb_checkpoints

* Update based on Kam's review

* Update to add a check if the env vars are set and add the option to set them
  • Loading branch information
dmsuehir authored and ashahba committed Oct 15, 2021
1 parent d2f3199 commit 60b1a0d
Show file tree
Hide file tree
Showing 4 changed files with 593 additions and 0 deletions.
2 changes: 2 additions & 0 deletions .gitignore
Expand Up @@ -14,3 +14,5 @@ output/
**/**.whl
**/**.tar.gz
tools/docker/models*
.ipynb_checkpoints

1 change: 1 addition & 0 deletions docs/notebooks/README.md
Expand Up @@ -8,3 +8,4 @@ The contributor is responsible for maintaining their notebooks.
| ------ | ------ | ------ |
| [perf_analysis](/docs/notebooks/perf_analysis) | Compare performance between stock and Intel Tensorflow for several models | [louie-tsai](https://github.com/louie-tsai), [ZhuoweiSi](https://github.com/ZhuoweiSi), [yinghu5](https://github.com/yinghu5), [charulatha27](https://github.com/charulatha27)|
| [ObjectDetection.ipynb](ObjectDetection.ipynb) | Visualize and benchmark the predictions of an object detection model served by TensorFlow Serving | [mhbuehler](https://github.com/mhbuehler) |
| [BERT Classifier fine tuning](/docs/notebooks/bert_classifier_fine_tuning/) | Fine tunes BERT base using the IMDB movie review dataset | [dmsuehir](https://github.com/dmsuehir) |

0 comments on commit 60b1a0d

Please sign in to comment.