随笔分类 -  pytorch

pytorch学习笔记
ntu120 数据理解
摘要:信息来源: https://medium.com/@lisajamhoury/understanding-kinect-v2-joints-and-coordinate-system-4f4b90b9df16 One of the most frequent questions I get from 阅读全文

posted @ 2020-08-22 09:22 wangling1820 阅读(578) 评论(0) 推荐(0)

pytorch中使用池化时输出向量的大小
摘要:注意:池化的各个函数中参数的stride默认设置为None,需要自己指定! 阅读全文

posted @ 2020-08-16 21:27 wangling1820 阅读(402) 评论(0) 推荐(0)

pytorch中softmax维度参数设置
摘要: 阅读全文

posted @ 2020-03-21 16:22 wangling1820 阅读(1369) 评论(0) 推荐(0)

RuntimeError: Expected object of scalar type Float but got scalar type Double for argument #4 'mat1'的解决方法
摘要:在pytorch中float32为float类型,而float64则为double类型,注意tensor的数据类型。 可以通过指定数据类型来获得所需要的类型数据。 torch.zeros([self.batch_size, self.num_layers, self.hidden_size], dt 阅读全文

posted @ 2020-03-20 18:15 wangling1820 阅读(9913) 评论(0) 推荐(0)

pytorch中的交叉熵(CrossEntropyLoss)的使用说明
摘要:官方示例: >>> loss = nn.CrossEntropyLoss() >>> input = torch.randn(3, 5, requires_grad=True) >>> target = torch.empty(3, dtype=torch.long).random_(5) >>> 阅读全文

posted @ 2020-03-20 18:07 wangling1820 阅读(1552) 评论(0) 推荐(0)

导航