0
点赞
收藏
分享

微信扫一扫

pytorch批量loss

guanguans 2022-04-30 阅读 50
        loss_char =torch.zeros(1).cuda()

        for j in range(len(restored)):
            loss_char=loss_char+criterion_char(restored[j], target)
举报

相关推荐

0 条评论