环境

  • python 3.9
  • numpy 1.24.1
  • pytorch 2.0.0+cu117

一、prune.random_unstructured

pytorch:文档地址

用法:

torch.nn.utils.prune.random_unstructured(module, name, amount)

参数:

  • module(torch.nn.Module) -包含要修剪的张量的模块

  • name(str) -module 中的参数名称,将对其进行修剪。

  • amount(<

05-19 06:12