Loss does not decrease during training (Word2Vec, Gensim)

后端 未结 2 1198
予麋鹿
予麋鹿 2021-01-06 05:42

What can cause loss from model.get_latest_training_loss() increase on each epoch?

Code, used for training:

class EpochSaver(CallbackA         


        
2条回答
  •  鱼传尺愫
    2021-01-06 06:04

    Up through gensim 3.6.0, the loss value reported may not be very sensible, only resetting the tally each call to train(), rather than each internal epoch. There are some fixes forthcoming in this issue:

    https://github.com/RaRe-Technologies/gensim/pull/2135

    In the meantime, the difference between the previous value, and the latest, may be more meaningful. In that case, your data suggest the 1st epoch had a total loss of 745896, while the last had (9676936-9280568=) 396,368 – which may indicate the kind of progress hoped-for.

提交回复
热议问题