Skip to content

janchk/NNVis

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

29 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

NNVis

Simple utils for neural network visualization


Currently weights distribution and activations distrubution are available.

API

# create your model
model = Model()

# put it in eval mode
model.eval()

# initialize visualizer object
nvis = NVIS()

# set model to Nvis
nvis.set_model(model)

# visualise weights distributions
nvis.plot_weights_ditributions()

# visualize activations distributions
nvis.plot_activations_distributions((<input_shape>))

The visualization will be located under vis/pdfs directory

About

Simple repository for visualizing different layer's activations distributions.

Resources

Stars

Watchers

Forks

Packages

No packages published

Languages