其他分享
首页 > 其他分享> > 自动随机逐层剪枝openpose

自动随机逐层剪枝openpose

作者:互联网

随机剪枝且保存模型

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