First I think there's a small error in the README tutorial: Should the correct directory should be data/labels instead of data/classes ?
Anyways, I get the following traceback:
/home/nic/.conda/envs/piwise/lib/python3.6/site-packages/torch/nn/functional.py:1423: UserWarning: nn.functional.upsample_bilinear is deprecated. Use nn.functional.upsample instead.
warnings.warn("nn.functional.upsample_bilinear is deprecated. Use nn.functional.upsample instead.")
/home/nic/machineLearning/Automatter/machine-learning/piwise/piwise/criterion.py:13: UserWarning: Implicit dimension choice for log_softmax has been deprecated. Change the call to include dim=X as an argument.
return self.loss(F.log_softmax(outputs), targets)
Traceback (most recent call last):
File "main.py", line 163, in <module>
main(parser.parse_args())
File "main.py", line 138, in main
train(args, model)
File "main.py", line 85, in train
board.image(color_transform(outputs[0].cpu().max(0)[1].data),
File "/home/nic/machineLearning/Automatter/machine-learning/piwise/piwise/transform.py", line 48, in __call__
color_image = torch.ByteTensor(3, size[1], size[2]).fill_(0)
IndexError: tuple index out of range
My first attempt at an obvious solution was to change it to
color_image = torch.ByteTensor(3, size[0], size[1]).fill_(0)
But I still have a size mismatch error. Tensor is 256 x 256 and Mask is 256.
First I think there's a small error in the README tutorial: Should the correct directory should be data/labels instead of data/classes ?
Anyways, I get the following traceback:
My first attempt at an obvious solution was to change it to
color_image = torch.ByteTensor(3, size[0], size[1]).fill_(0)But I still have a size mismatch error. Tensor is 256 x 256 and Mask is 256.