self.conv1 = nn.Conv1d(512, 32, kernel_size=3, padding=1)怎么设置权重类型转换为cuda.DoubleTensor
时间: 2024-05-09 22:20:23 浏览: 88
对tensorflow中tf.nn.conv1d和layers.conv1d的区别详解
可以使用`.double()`方法将权重转换为`cuda.DoubleTensor`类型,如下所示:
```
self.conv1 = nn.Conv1d(512, 32, kernel_size=3, padding=1).double().cuda()
```
这将把`conv1`的权重转换为`cuda.DoubleTensor`类型,并将它们移动到GPU上。
阅读全文