yolo 自带的模型大小减半方法

最新推荐文章于 2024-07-29 17:20:55 发布

小女孩真可爱

最新推荐文章于 2024-07-29 17:20:55 发布

阅读量468

点赞数

分类专栏：便捷小代码文章标签： pytorch 深度学习人工智能

本文链接：https://blog.csdn.net/m0_48095841/article/details/124847096

版权

便捷小代码专栏收录该内容

14 篇文章 0 订阅

订阅专栏

def strip_optimizer(path, f='weights/best.pt', s=''):  # from yolo_utils.general import *; strip_optimizer()
    # Strip optimizer from 'f' to finalize training, optionally save as 's'
    x = torch.load(f, map_location=torch.device('cpu'))

    for key in 'optimizer', 'training_results', 'wandb_id':
        x[key] = None
    x['epoch'] = -1
    x["model"].half()  # to FP16
    for p in x["model"].parameters():
        p.requires_grad = False
    # torch.save(x, s or f)
    torch.save(x, path)
    # mb = os.path.getsize(s or f) / 1E6  # filesize
    # print('Optimizer stripped from %s,%s %.1fMB' % (f, (' saved as %s,' % s) if s else '', mb))