SimCSE

SimCSE copied to clipboard

期待能出个基于simcse的中文预训练模型

3

lonngxiang

694344851

在计算output pooling时不用mask吗？

1

if pooling == 'first-last-avg': outputs = [ keras.layers.GlobalAveragePooling1D()(outputs[0]), keras.layers.GlobalAveragePooling1D()(outputs[-1]) ] output = keras.layers.Average()(outputs) elif pooling == 'last-avg': output = keras.layers.GlobalAveragePooling1D()(outputs[-1]) elif pooling == 'cls': output = keras.layers.Lambda(lambda x: x[:, 0])(outputs[-1])...

shaomai00

怎么修改batch_size

1


yuxuan2015