Anomaly-Transformer (ICLR 2022 )代码通过CPU复现

代码地址：GitHub - thuml/Anomaly-Transformer: About Code release for "Anomaly Transformer: Time Series Anomaly Detection with Association Discrepancy" (ICLR 2022 Spotlight), https://openreview.net/forum?id=LzQQ89U1qm_

尝试在CPU上复现MSL的结果

遇到两个报错：

1 Torch not compiled with CUDA enabled”

解决办法：

在程序最开始的地方加上

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
使用的时候

self.encoder = EncoderAtt(input_size=self.X.shape[1], hidden_size=encoder_hidden_size, T=T).to(device)
就是在原来.cuda()的地方都改成.to(device)
参考链接：https://blog.csdn.net/ssjdoudou/article/details/103640129

2 ValueError: too many values to unpack (expected 2)

原因是没有输入参数，解决可参考Anomaly-Transformer (ICLR 2022 )代码复现_hhx_1220的博客-CSDN博客

以下是跑出的结果：

 基本和作者文中的差不多：