torch.nn.utils.prune.remove

torch.nn.utils.prune.remove(module, name) [исходный код]

Удаляет параметризацию прореживания из модуля и метод прореживания из обработчика прямого вызова.

Прореженный параметр с именем name остается прореженным навсегда, а параметр с именем name+'_orig' удаляется из списка параметров. Аналогично, буфер с именем name+'_mask' удаляется из списка буферов.

Примечание

Само прореживание НЕ отменяется и не обращается вспять!

Параметры:

• module (nn.Module) – модуль, содержащий тензор для прореживания
• name (str) – имя параметра в module, к которому будет применяться прореживание.

Примеры

>>> m = random_unstructured(nn.Linear(5, 7), name="weight", amount=0.2)
>>> m = remove(m, name="weight")