Using LogSoftMax with multiple labels

210 views
Skip to first unread message

UserNotDefined

unread,
Apr 24, 2015, 7:16:00 PM4/24/15
to tor...@googlegroups.com
How would I use LogSoftMax with multiple labels?

In the example here: 

as far as I understand the data_outputs are only 1 dimensional, i.e. from 1-3

I changed it to work with my own data with 1 label, but I can't seem to pinpoint where to change it to allow multiple labels.

soumith

unread,
Apr 26, 2015, 11:52:42 PM4/26/15
to torch7 on behalf of UserNotDefined
With multiple labels per sample, you would have nn.Sigmoid as the last layer of your network and use nn.BCECriterion as your criterion.

--
You received this message because you are subscribed to the Google Groups "torch7" group.
To unsubscribe from this group and stop receiving emails from it, send an email to torch7+un...@googlegroups.com.
To post to this group, send email to tor...@googlegroups.com.
Visit this group at http://groups.google.com/group/torch7.
For more options, visit https://groups.google.com/d/optout.

Reply all
Reply to author
Forward
0 new messages