# 训练模型loss_list = for epoch in range100 running_loss = 00 for i in rangelenX_train inputs = torchFloatTensorX_trainiunsqueeze0 targets = torchFloatTensory_traini optimizerze
这个警告提示出现的原因是目标张量与输出张量的形状不一致,需要将目标张量的形状与输出张量保持一致。
解决方法:
在将目标张量传递给损失函数之前,将其形状调整为与输出张量一致的形状,例如:
targets = targets.view(outputs.shape)
或者
targets = targets.reshape(outputs.shape)
这将确保目标张量与输出张量具有相同的形状,从而避免出现警告。
原文地址: https://www.cveoy.top/t/topic/OF9 著作权归作者所有。请勿转载和采集!