这个警告提示出现的原因是目标张量与输出张量的形状不一致,需要将目标张量的形状与输出张量保持一致。

解决方法:

在将目标张量传递给损失函数之前,将其形状调整为与输出张量一致的形状,例如:

targets = targets.view(outputs.shape)

或者

targets = targets.reshape(outputs.shape)

这将确保目标张量与输出张量具有相同的形状,从而避免出现警告。

# 训练模型loss_list = for epoch in range100 running_loss = 00 for i in rangelenX_train inputs = torchFloatTensorX_trainiunsqueeze0 targets = torchFloatTensory_traini optimizerze

原文地址: https://www.cveoy.top/t/topic/OF9 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录