torch.nn.utils.prune.remove

torch.nn.utils.prune.remove(module, name) [source]

Удаляет перепараметризацию обрезки из модуля и метод обрезки из крючка переадресации. Обрезанный параметр с именем 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')