ArthurWish / DCTLN-

DCTLN复现

运行环境

需要开源的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: 训练图片保存