Skip to content

Latest commit

 

History

History
32 lines (21 loc) · 769 Bytes

File metadata and controls

32 lines (21 loc) · 769 Bytes

Model Inpector

This repo contains a tool to visualize your models and all the hidden layers.
You can inspect the feature maps and all the layers output in real time.

Quickstart

There are 2 ways of using this tool, from your program or as CLI.

Using python

import modelinspector

model = ...
x = ... # 1 element batched tensor

inspector = modelinspector.ModelInspector(model, x)

inspector.run() 

You will then open your browser (normally at 127.0.0.1:8050 unless changed config).

CLI

python -mmodelinspector <path_to_model> <path_to_saved_tensor>

Screenshots

inspecting feature maps inspecting linear layers