pyTorch
一只努力的喵
这个作者很懒,什么都没留下…
展开
-
IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python
IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number错误分析 错误原因:pytorch版本不同导致 train_loss += loss.data[0]是是pytorch0.3.1版本代码,在0.5以上版本中就会报错 解决办法 train_loss += loss.data[0] 修改为 train_loss += loss.item() ...原创 2020-06-09 18:28:15 · 376 阅读 · 0 评论 -
pyTorch中模型加载
pyTorch框架中的torchvision 在pyTorch框架中,torchvision是一个非常重要的包,其中包含三部分:torchvision.datasets、torchvision.models、torchvision.transforms。 详细介绍参考官网和源码 pyTorch中的预训练模型构建和下载 以vgg16为例: 预训练模型中默认输入RGB图像,h和w不低于224,图像的像...原创 2019-12-30 15:40:26 · 181 阅读 · 0 评论