Total imgs for test of occ_level ZERO and occ_type 4341
Testing
0%| | 0/4341 [00:00<?, ?it/s]
Traceback (most recent call last):
File "/home/CompositionalNets/Code/test.py", line 138, in <module>
acc,scores = test(models=nets, test_data=test_imgset, batch_size=1)
File "/home/CompositionalNets/Code/test.py", line 40, in test
output, *_ = models[0](input)
File "/home/toshi/base/lib/python3.9/site-packages/torch/nn/modules/module.py", line 889, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/CompositionalNets/Code/model.py", line 62, in forward
mix_likeli = self.pointwiseinference(vmf_activations)
File "/home/toshi/base/lib/python3.9/site-packages/torch/nn/modules/module.py", line 889, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/CompositionalNets/Code/model.py", line 223, in forward
foreground = foreground.reshape((-1, self.num_classes,
RuntimeError: shape '[-1, 10, 4, 1, 7, 12]' is invalid for input of size 3612
import os, sys
p = os.path.abspath('.')
sys.path.insert(1, p)
imgset = Imgset(imgs, masks, labels, imgLoader, bool_square_images=False,bool_cutout=False,bool_pytorch=bool_pytorch)
imgset = Imgset(imgs, masks, labels, imgLoader, bool_square_images=False)