laymon.monitoring

Module Contents

Classes

FeatureMapMonitoring()

A wrapper class for adding a model or model layers for monitoring the feature maps

class laymon.monitoring.FeatureMapMonitoring

Bases: object

A wrapper class for adding a model or model layers for monitoring the feature maps during training of the model.

add_layer(self, layer, layer_name)

Adds the layer whose feature maps are to be monitored. :param layer: pyTorch layer :param layer_name: (str) name of the layer :return: the observer object of the layer being monitored

_add_layer(self, layer, layer_name)

Create a observer class of the layer to be monitored and adds it to the list of observers being monitored. :param layer: pyTorch layer :param layer_name: (str) name of the layer :return: Observer object of the layer

_remove_layer(self, layer_name)

Removes an observer from the list of observers being monitored. :param layer_name: name of the observer :return: True if observer was deleted, else False

remove_layer(self, layer_name)

Remove the layer from list of layer being monitored. :param layer_name: str (name of the layer) :return: True if the layer is deleted from the list of monitored objects, else False

add_model(self, model)

Registers all the layers a pyTorch model whose activations maps are to monitored. :param model: :return:

start(self)

Starts monitoring the feature maps of the registered layers/model.