赞
踩
UserWarning: Using a target size (torch.Size([256, 1])) that is different to the input size (torch.Size([256, 1, 1])). This will likely lead to incorrect results due to broadcasting. Please ensure they have the same size.
使用L1正则化的问题。
Pytorch遇到的坑:为什么模型训练时,L1loss损失无法下降?_损失降不下去_一个菜鸟的奋斗的博客-CSDN博客
参考修改后,警告消失,损失大幅下降
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。