How can i use MSE criterion?

[neuralix]

Running examples/mnist.lua simply, i can't change criterion to MSE Criterion.
Can you tell me the how-to?

[lvdmaaten]

Sorry but this is for reporting issues in torchnet, not for general Torch questions.

https://github.com/torch/nn/blob/master/doc/criterion.md#msecriterion

[neuralix]

I think the problem is still exist.

[Kaixhin]

Yes, you cannot just use any criterion. The MNIST example is for classification, whereas MSE should be used for regression. This can be found in the Torch nn documentation.

[neuralix]

Thank you.

[neuralix]

I suspect returning vector in getIterator(mode) {...} is not work.
(specifically, variable 'load' in ListDataset cannot contain vector as target.

target = torch.LongTensor(1,10) or target = torch.LongTensor(10, 1)
(instead of original line,
target = torch.LongTensor{dataset.label[idx] + 1} )
make error.

Will you have a plan to enhance it near some day?