论文复现---BeatGAN
原创 小王搬运工 时序课堂 2024-06-11 10:02 四川
BeatGAN: Anomalous Rhythm Detection using Adversarially Generated Time Series
https://github.com/Vniex/BeatGAN
-
PyTorch (1.0.0)
-
scikit-learn (0.20.0)
-
biosppy (0.6.1) # For data preprocess
-
tqdm (4.28.1)
-
matplotlib (3.0.2)
注意事项:
问题1:
这个地方按照这个安装我尝试了很多次都没安装成功,如果配置过国内镜像源如清华源,其中是没有老版本pytorch,清华源最早支持gpu的pytorch版本是1.5.0。而且torchvision的0.2.1这个版本即使是在pytorch自己的仓库都找不到,只能找到0.2.0,更别说在国内的镜像源中了。
解决方法:
去官网下载安装包进行安装:
https://download.pytorch.org/whl/cu100/torch_stable.html
然后找到torchvision,可以发现没有torch0.2.1,只有0.2.0,这里按照0.2.0安装也没有问题。
问题2:python版本不适配
问题3:GPU无法使用
这里我更改GPU检测部分,默认使用CPU运行一切正常!
其余的运行时提示缺少什么功能库,按照最新版默认安装即可。
https://www.dropbox.com/scl/fo/c14su187vh95gzchm7fgk/AGGykb6I5hMrbazgC8TDvPU?rlkey=rbzdfgp5y6bftpuwueqg7cpb1&e=1&dl=0
下载路径:BeatGAN\experiments\ecg\dataset\preprocessed\ano0
修改数据集加载路径:
本地电脑可以正常运行,由于性能原因未等到训练完,等待放到服务器上运行。
全套代码分享,包含数据集等,配好环境,没有问题即可运行。
链接:https://pan.baidu.com/s/1FUP-P6S4qm9WQZOPTJ3G-A
提取码:irb7
--来自百度网盘超级会员V6的分享
标签:安装,0.2,---,BeatGAN,pytorch,复现,https,com From: https://blog.csdn.net/qq_25368751/article/details/139591386