Nlp项目实战自定义模板框架

1.utils模块

一般存放一些辅助函数,其他包不能放的都可以放到这个包里面

2.trainer模块

训练模块

3.model模块

loss.py:损失函数
metric.py:结果展示
model.py:自定义模型

4.logger模块

日志模块主要是记录模型训练,网络收敛的一个情况.以及可视化展示(tensorboard工具).

5.data_loader模块

数据读取模块,一个是数据预处理功能,一个数据采样功能.

6.data模块

数据模块存放着我们需要训练的数据.

7.base模块

base模块里面存放这pytorch框架通用的模块.

8.test.py,train.py

测试和训练的入口文件