python
1 2for i, data in enumerate(train_loader,0): 3 optimizer.zero_grad() 4 inputs, labels = data[0].to(device), data[1].to(device) 5 outputs = net(inputs) 6 mu = outputs[0][:3] 7 eye = Variable(torch.eye(3),requires_grad=True).to(device) 8 dist = tdist.multivariate_normal.MultivariateNormal(mu, eye) 9 train_loss = -dist.log_prob(labels[0]) 10 train_loss.backward() 11 optimizer.step() 12 13<変数内容> 14(eye) 15tensor([[1., 0., 0.], 16 [0., 1., 0.], 17 [0., 0., 1.]], device='cuda:0', grad_fn=<CopyBackwards>) 18 19(mu) 20tensor([-0.1601, 0.1404, 0.4074], device='cuda:0', grad_fn=<SliceBackward>)
pytorchで多次元ガウス分布の対数計算(multivariate_normal.MultivariateNormal)を試したところ、以下のようなエラーが出て計算を実行することができません。muを平均、eyeを共分散行列として設定しています。
エラー内容、発生している原因を教えていただけると幸いです。
self._unbroadcasted_scale_tril = torch.cholesky(self.covariance_matrix)
RuntimeError: CUDA error: invalid argument
あなたの回答
tips
プレビュー