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
测试和训练的入口文件