pytorch只计算20%代码如何编写

最新推荐文章于 2024-07-30 07:00:00 发布

zouxiaolv

最新推荐文章于 2024-07-30 07:00:00 发布

阅读量125

点赞数

分类专栏： python pytorch 文章标签： python

本文链接：https://blog.csdn.net/zouxiaolv/article/details/108943046

版权

模型评估批量预测验证数据模型损失多个人体姿势估计

关键词由CSDN通过智能技术生成

pytorch 同时被 2 个专栏收录

97 篇文章 4 订阅

订阅专栏

python

95 篇文章 2 订阅

订阅专栏

https://github.com/hellojialee/Multi-Person-Pose-using-Body-Parts/blob/7dd783f8dca4a8292232fcc724ecbd911a69f428/training/train_common.py#L82

  model_metrics = []
    inhouse_metrics = []
    t0 = time()
    for i in range(validation_steps):  # 分成很多个batch进行预测估计的，为了减少validation耗时，在计算validation部分数据
        # validation_steps　＝　val_samples//batch_size 为了防止内存OOM，所以要分batch预测
        # if random.randint(0, 9) < 5:  # 只计算20%的数据
        #     continue
        X, GT = next(val_thre.get())
        Y = multi_model.predict(X)

        model_losses = [(np.sum((gt - y) ** 2) / gt.shape[0] / 2) for gt, y in zip(GT,Y)]
        # 与模型定义时的loss保持一致，除以2的好处是，平方项的微分会出现2，抵消，可以减少乘法操作
        mm = sum(model_losses)