pytorch
wchzh2015
这个作者很懒,什么都没留下…
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
pytorch 的 nelement()
pytorch中的 nelement() 可以统计 tensor (张量) 的元素的个数。 输入代码:环境 pytorch 1.0 ,Python 3.6 import torch x = torch.randn(size = (4,3,5,6)) x = torch.tensor([[1.], [2], [3]]) print(x.nelement()) 结果是: 3 如果有误,...原创 2018-12-15 10:38:20 · 11067 阅读 · 0 评论 -
Pytorch:Unexpected key(s) in state_dict:
实验环境 pytorch 0.4 Python 3.6 在使用交叉验证后,将保存的模型重新加载求平均的结果。但是出现了以下的错误。 实验的bug截图 在此感谢这篇博文:https://blog.youkuaiyun.com/jacke121/article/details/82495184 自己修改后的关键代码如下: model_resnet101 = get_net...原创 2018-12-25 21:06:04 · 16756 阅读 · 0 评论 -
Pytorch 常见错误
RuntimeError: bool value of Tensor with more than one value is ambiguous 引发的原因:少加了一个括号,CE_Net_ 是一个类, 改正后:原创 2019-06-27 18:11:26 · 1216 阅读 · 0 评论 -
使用 清华源 安装pytorch
python3 -m pip install --upgrade torch torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple原创 2019-06-27 19:00:37 · 8201 阅读 · 0 评论
分享