pytorch
一舫
我一定要回家!
展开
-
torch.nn 在 Pycharm 中无法自动提示
torch 1.6.0 的坑,更新到 1.10.0 可以解决原创 2021-12-04 16:19:33 · 774 阅读 · 0 评论 -
Pytorch Batch Normalization 中 training与track_running_stats问题
一般来说,training与track_running_stats有四种组合。training=True,track_running_stats=True,这个是期望中的训练阶段的设置,此时BN将会跟踪整个训练过程中batch的统计特性。 training=True,track_running_stats=False,此时BN只会计算当前输入的训练batch的统计特性,可能没法很好地描述全局的数据统计特性。 training=False,track_running_stats=True,这个是期望中转载 2020-09-23 11:09:19 · 1367 阅读 · 0 评论 -
RuntimeError: expected device cpu but got device cuda:0
RuntimeError: expected device cpu but got device cuda:0input = input.cuda()RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the samemodel.cuda()原创 2020-09-03 11:08:59 · 1213 阅读 · 0 评论 -
RuntimeError: Exporting the operator repeat_interleave to ONNX opset version 9 is not supported.
RuntimeError: Exporting the operator repeat_interleave to ONNX opset version 9 is not supported. Please open a bug to request ONNX export support for the missing operator.pytorch转onnx报的错,原因是模型中的repeat_interleave操作在onnx中还不支持,可以把这个操作换成别的,emmm.....,我不知道换成.原创 2020-08-27 21:45:52 · 4644 阅读 · 2 评论 -
Windows10下安装Anaconda、Tensorflow-GPU2.0、Pytorch
1.安装Anacondahttps://www.anaconda.com/products/individual原创 2020-05-25 21:38:25 · 420 阅读 · 0 评论