需要开源的easydl https://github.com/thuml/easydl
需要pytorch
下载包 使用指令 pip install -r requirements.txt
data.py 数据读取、处理、加噪
network.py 网络模型
train.py 训练模型,运行时只需要运行该文件
utils.py 工具类
train_log.yaml log日志配置代码
CWRU: 凯斯西楚大学数据集
Log: 日志
SaveModel: 模型保存
SavePicture: 训练图片保存