随机剪枝且保存模型
import osimport torchfrom models.CocoPoseNet import CocoPoseNetfrom new_openpose_prune_model import CocoPoseNet as new_netimport randomclass PruneNet():
def __init__(self):
self.weight_dat
标签:__,剪枝,CocoPoseNet,逐层,new,import,openpose
来源: https://blog.51cto.com/u_15177056/2725905