随机剪枝且保存模型
import osimport torchfrom models.CocoPoseNet import CocoPoseNetfrom new_openpose_prune_model import CocoPoseNet as new_netimport randomclass PruneNet(): def __init__(self): self.weight_dat
相关文章
- 03-24自动随机逐层剪枝openpose
2024-03-24 18:24:28
随机剪枝且保存模型
import osimport torchfrom models.CocoPoseNet import CocoPoseNetfrom new_openpose_prune_model import CocoPoseNet as new_netimport randomclass PruneNet(): def __init__(self): self.weight_dat