Gensim word2vec训练在批处理结束时不回调

我有兴趣在Gensim word2vec模型上放置一个回调,以便在每个批处理后触发一些函数。每 documentation ,可以在批处理结束或纪元结束时放置回调。然而,如下面的MVE所示,实际上只有纪元回调会触发。

要运行示例,让 corpus_filepath 直接指向一个以行分隔的未加标点的句子文件(给定行上的句子中的单词应以空格分隔)。您可能还需要更改 workers 在 Word2Vec 实例化。

from gensim.models import Word2Vec
from gensim.models.callbacks import CallbackAny2Vec

corpus_filepath = 'train.txt'
out_filepath = 'out.txt'

class MyCallback(CallbackAny2Vec):
    def __init__(self):
        pass

    def on_batch_end(self, model):
        print('batch end')

    def on_epoch_end(self, model):
        print('epoch end')


callback = MyCallback()
model = Word2Vec(size=300, window=5, min_count=0, workers=64)
print('Making vocabulary...')
model.build_vocab(corpus_file=corpus_filepath)
print('Beginning training...')
model.train(corpus_file=corpus_filepath, epochs=5, total_words=model.corpus_total_words, callbacks=[callback])

输出不正确(缺少批打印输出):

Making vocabulary...
Beginning training...
epoch end
epoch end
epoch end
epoch end
epoch end

我做错了什么?