I wanted to try working on CIFAR10, I modified the channel number =3 in convLayer e the kernel dim = 24 in convLayer. But it is not working in the primaryCaps u = u.view(x.size(0), 32 6 6, -1) that is giving to me errot :
ipython-input-4-7b4e2b87bd5c> in forward(self, x)
14 print( "PrimaryCaps {}".format(x.size()))
15 #u = u.view(x.size(0), 32 6 6, -1)
---> 16 u = u.view(x.size(0), 32 6 6, -1)
17 return self.squash(u)
18
RuntimeError: invalid argument 2: size '[100 x 1152 x -1]' is invalid for input with 204800 elements at /pytorch/aten/src/TH/THStorage.c:37
I wanted to try working on CIFAR10, I modified the channel number =3 in convLayer e the kernel dim = 24 in convLayer. But it is not working in the primaryCaps u = u.view(x.size(0), 32 6 6, -1) that is giving to me errot :
ipython-input-4-7b4e2b87bd5c> in forward(self, x) 14 print( "PrimaryCaps {}".format(x.size())) 15 #u = u.view(x.size(0), 32 6 6, -1) ---> 16 u = u.view(x.size(0), 32 6 6, -1) 17 return self.squash(u) 18 RuntimeError: invalid argument 2: size '[100 x 1152 x -1]' is invalid for input with 204800 elements at /pytorch/aten/src/TH/THStorage.c:37
Do you have any advise for me?