search

koomri / text-segmentation

Implementation of the paper: Text Segmentation as a Supervised Learning Task
253 stars 57 forks source link

Why is softmax not used in training? #16

Closed MuhammadAhmed8 closed 2 years ago

MuhammadAhmed8 commented 2 years ago

In the train function, softmax activation is not applied but it is applied in the validate function. Why is that so? Thanks.