Модуль: tf.contrib.model_pruning

Реализация обрезки моделей в tensorflow.

Классы

class MaskedBasicLSTMCell: Базовая ячейка LSTM с обрезкой.

class MaskedLSTMCell: Ячейка LSTM с обрезкой.

class Pruning

Функции

apply_mask(...): Применить маску к заданному весовому тензору.

get_masked_weights(...)

get_masks(...)

get_pruning_hparams(...): Получить объект tf.HParams с параметрами по умолчанию.

get_thresholds(...)

get_weight_sparsity(...): Получить разреженность весов.

get_weights(...)

graph_def_from_checkpoint(...): Преобразует данные контрольной точки в GraphDef.

masked_conv2d(...): Добавляет 2D свертку, за которой необязательно следует слой batch_norm.

masked_convolution(...): Добавляет 2D свертку, за которой необязательно следует слой batch_norm.

masked_fully_connected(...): Добавляет разреженный полносвязный слой. Матрица весов маскируется.

strip_pruning_vars_fn(...): Удаляет переменную маски из графа.

train(...): Обёртка вокруг функции tf-slim's train.