Skip to content

Latest commit

 

History

History
executable file
·
52 lines (39 loc) · 2.74 KB

README.md

File metadata and controls

executable file
·
52 lines (39 loc) · 2.74 KB
Raw

Run Intel® AI Reference Models in a Jupyter Notebook

This Jupyter notebook helps you choose and run a comparison between two models from the Intel® AI Reference Models repo using Intel® Optimizations for TensorFlow*. When you run the notebook, it installs required package dependencies, displays information about your platform, lets you choose the two models to compare, runs those models, and finally displays a performance comparison chart.

Supported Models

| Model | Framework | Mode | Supported Precisions | | ----- | --------- | ---- | ----------- | -------------------- | | ResNet50v1.5 | TensorFlow | Inference | FP32, BFloat32, FP16, INT8, BFloat16 | | BERT Large | TensorFlow | Inference | FP32, BFLOAT16, FP16

Environment Setup

Instead of installing or updating packages system-wide, it's a good idea to install project-specific Python packages in a Python virtual environment localized to your project. The Python virtualenv package lets you do just that. Using virtualenv is optional, but recommended.

The jupyter notebook runs on Ubuntu distribution for Linux.

  1. Virtualenv Python Environment Install virtualenv on Ubuntu using these commands:

    sudo apt-get update
sudo apt-get install python-dev python-pip
sudo pip install -U virtualenv  # system-wide install

    Activate virtual environment using the following command:

    virtualenv -p python ai_ref_models
source ai_ref_models/bin/activate

  2. Jupyter Notebook Support:

    Install Jupyter notebook support with the command:

       pip install notebook

    Refer to the Installing Jupyter guide for details.

How to Run the Notebook

  1. Clone the Intel® AI Reference Models repo: 
    git clone https://github.com/IntelAI/models.git
  2. Launch the Jupyter notebook server: jupyter notebook --ip=0.0.0.0
  3. Follow the instructions to open the URL with the token in your browser, something like this: http://127.0.0.1:8888/tree?token=<token>
  4. Browse to the models/notebooks/ folder
  5. Click the AI_Reference_Models notebook file - AI_Reference_Models.ipynb.
  6. Read the instructions and run through each notebook cell, in order, ending with a display of the analysis results. Note that some cells prompt you for input, such as selecting the model number you'd like to run.
  7. When done, you should deactivate the virtualenv, if you used one, with the command: deactivate