/home/raul/Desktop/CapsNet-pytorch/net.py:28: UserWarning: Implicit dimension choice for softmax has been deprecated. Change the call to include dim=X as an argument.
c = F.softmax(self.b)
/home/raul/Desktop/CapsNet-pytorch/net.py:38: UserWarning: Implicit dimension choice for softmax has been deprecated. Change the call to include dim=X as an argument.
c = F.softmax(b_batch.view(-1, output_caps)).view(-1, input_caps, output_caps, 1)
Traceback (most recent call last):
File "/home/raul/Desktop/CapsNet-pytorch/net.py", line 275, in
train(epoch)
File "/home/raul/Desktop/CapsNet-pytorch/net.py", line 244, in train
batch_idx / len(train_loader), loss.data[0]))
IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number
/home/raul/Desktop/CapsNet-pytorch/net.py:28: UserWarning: Implicit dimension choice for softmax has been deprecated. Change the call to include dim=X as an argument. c = F.softmax(self.b) /home/raul/Desktop/CapsNet-pytorch/net.py:38: UserWarning: Implicit dimension choice for softmax has been deprecated. Change the call to include dim=X as an argument. c = F.softmax(b_batch.view(-1, output_caps)).view(-1, input_caps, output_caps, 1) Traceback (most recent call last): File "/home/raul/Desktop/CapsNet-pytorch/net.py", line 275, in
train(epoch)
File "/home/raul/Desktop/CapsNet-pytorch/net.py", line 244, in train