Fork me on GitHub

随笔分类 -  Pytorch

摘要:使用pytorch的小伙伴们,一定看过下面这段代码 _, predicted = torch.max(outputs.data, 1) 那么,这里的 下划线_ 表示什么意思? 首先,torch.max()这个函数返回的是两个值,第一个值是具体的value(我们用下划线_表示),第二个值是value所 阅读全文
posted @ 2021-02-25 16:19 stardsd 阅读(3913) 评论(0) 推荐(0) 编辑
摘要:N: batch; C: channel H: height W: width Caffe 的Blob通道顺序是:NCHW; Tensorflow的tensor通道顺序:默认是NHWC, 也支持NCHW,使用cuDNN会更快; Pytorch中tensor的通道顺序:NCHW TensorRT中的t 阅读全文
posted @ 2021-02-25 15:16 stardsd 阅读(2576) 评论(0) 推荐(0) 编辑