自然言語処理でGPUを使用するとRuntimeError: expected device cuda:0 but got device cpuというエラーが出てしまいまいます。試したことはこのエラーを検索しましたがいろいろ出てきましたが、どこのコードを改善するかわかりませんでした。
Python
1class PositionalEncoder(nn.Module): 2 def __init__(self, d_model=300, max_seq_len=256): 3 super().__init__() 4 5 self.d_model = d_model 6 7 pe = torch.zeros(max_seq_len, d_model) 8 9 for pos in range(max_seq_len): 10 for i in range(0, d_model, 2): 11 pe[pos, i] = math.sin(pos / (10000 ** ((2 * i)/d_model))) 12 pe[pos, i + 1] = math.cos(pos / 13 (10000 ** ((2 * i)/d_model))) 14 self.pe = pe.unsqueeze(0) 15 self.pe.requires_grad = False 16 17 def forward(self, x): 18 ret = math.sqrt(self.d_model)*x + self.pe 19 return ret
あなたの回答
tips
プレビュー