Open yxbu opened 4 years ago
Hi, It may be due to version changes. I have no problem with PyTorch 1.1.0.
It is possible to solve the problem without degrade the pytorch?
Yes, but unfortunately I am currently so busy. If you can send a PR it is really appreciated.
Yes, but unfortunately I am currently so busy. If you can send a PR it is really appreciated.
Thanks for your kind reply. I have solved the problem by replace the "Q = max_mask.diag().float()" with "Q = max_mask.float().diag()."
Hello, the link to download the data set cannot be opened. Is there any solution?
I get a runtime error when I run demo.py. the version of my Pytorch is 1.2.0.what's the problem? 1747 371
Epoch: 1 0it [00:00, ?it/s]Traceback (most recent call last): File "demo.py", line 133, in
train_loss, train_acc = train(epoch)
File "demo.py", line 73, in train
loss.backward()
File "/home/yxbu/.conda/envs/torch/lib/python3.7/site-packages/torch/tensor.py", line 118, in backward
torch.autograd.backward(self, gradient, retain_graph, create_graph)
File "/home/yxbu/.conda/envs/torch/lib/python3.7/site-packages/torch/autograd/init.py", line 93, in backward
allow_unreachable=True) # allow_unreachable flag
File "/home/yxbu/.conda/envs/torch/lib/python3.7/site-packages/torch/autograd/function.py", line 77, in apply
return self._forward_cls.backward(self, *args)
File "/home/yxbu/code/spdnet/spdnet/spd.py", line 296, in backward
Q = max_mask.diag().float()
RuntimeError: _th_diag not supported on CUDAType for Bool
0it [00:00, ?it/s]