Appropriate loss function in pytorch when output is an array of float numbers

526 views Asked by At

I am writing an encoder/decoder model very similar to https://pytorch.org/tutorials/intermediate/seq2seq_translation_tutorial.html

The only difference is, here, the words are represented by some indices. I want to show them based on another metric, which are represented by flaot numbers.

The loss function nn.criterion = nn.NLLLoss(), seems to be working for times we are only workin with classes.

If my output array is not an array of integers, but an array of float numbers, what kind of loss function I can use? Considering all other parts are similar to the tutorial?

Thanks in advance.

0

There are 0 answers