Concise_SimCSE

A concise implementation of SimCSE
模型SimCSE的简洁实现

一、简介

提供一个代码结构清晰、简洁易懂的SimCSE实现。

二、代码结构

data：包含训练数据；
config.py：各类参数统一定义在该文件中；
model.py：模型结构定义在该文件中；
train.py：模型训练的流程和入口在该文件中；
utils.py：各类数据加载的函数和全局变量定义在该文件中；
notebook.ipynb：notebook版本的代码；

三、运行代码

python train.py