2024年1月12日
Pytorch中的交叉熵CE和均方误差MSE分别是如何计算的？
摘要： 本文主要关注输入输出的形状，通过两种标签形式探索一下其计算机制。 交叉熵损失函数 实验设置：假设采用AutoEncoder+分类器。AE负责重构图像，计算MSE。分类器通过latent vector计算23个类别的概率向量。 import torch nn as nn net = AutoEncod 阅读全文
posted @ 2024-01-12 14:53 倦鸟已归时 阅读(41) 评论(0) 推荐(0) 编辑