Concise_SimCSE

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

一、简介

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

二、代码结构

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

三、运行代码

python train.py