Visualize your model with Model Explorer

What is Model Explorer?

Model Explorer is a visualization tool for inspecting neural network structures and execution graphs. Arm provides a VGF adapter for Model Explorer, allowing you to visualize .vgf models created from your training and export pipeline.

This lets you inspect model architecture, tensor shapes, and graph connectivity before deployment. This can be a powerful way to debug and understand your exported neural graphics models.

Set up the VGF adapter

The VGF adapter extends Model Explorer to support .vgf files exported from the Model Gym toolchain.

Install the VGF adapter with pip

Run:

    

        
        
pip install vgf-adapter-model-explorer

    

The VGF adapter model explorer source code is available on GitHub .

Install Model Explorer

The next step is to make sure the Model Explorer itself is installed. Use pip to set it up:

    

        
        
pip install torch ai-edge-model-explorer

    

Launch the viewer

Once installed, launch the explorer with the VGF adapter:

    

        
        
model-explorer --extensions=vgf_adapter_model_explorer

    

Use the file browser to open the .vgf model exported earlier in your training workflow.

Continue to the next section to learn about bringing your own use-cases to the Model Gym.