Skip to content

Commit 6b09113

Browse files
Sunting78Sunting78
authored
add amp and to_static (#542)
1 parent 7dc02c8 commit 6b09113

28 files changed

+609
-223
lines changed

‎configs/anomaly/anomaly_base.yaml

+8-12
Original file line numberDiff line numberDiff line change
@@ -7,27 +7,23 @@ epoch: 1
77
training: True
88
anomaly_ratio: 1
99
task: anomaly
10+
to_static_train: False
11+
use_amp: False
12+
amp_level: O2
1013

1114

1215
dataset:
1316
name: TSADDataset
14-
dataset_root: ./data/
15-
train_path: ./data/swat_train.csv
16-
val_path: ./data/swat_test.csv
17+
dataset_root: ./dataset/
18+
train_path: ./dataset/ts_anomaly_examples/train.csv
19+
val_path: ./dataset/ts_anomaly_examples/test.csv
1720
scale: True
1821
time_feat: False
1922

2023
info_params:
2124
freq: 1
22-
label_col: "Normal/Attack"
23-
feature_cols: ['FIT101', 'LIT101', ' MV101', 'P101', 'P102', ' AIT201', 'AIT202',
24-
'AIT203', 'FIT201', ' MV201', ' P201', ' P202', 'P203', ' P204',
25-
'P205', 'P206', 'DPIT301', 'FIT301', 'LIT301', 'MV301', 'MV302',
26-
' MV303', 'MV304', 'P301', 'P302', 'AIT401', 'AIT402', 'FIT401',
27-
'LIT401', 'P401', 'P402', 'P403', 'P404', 'UV401', 'AIT501',
28-
'AIT502', 'AIT503', 'AIT504', 'FIT501', 'FIT502', 'FIT503',
29-
'FIT504', 'P501', 'P502', 'PIT501', 'PIT502', 'PIT503', 'FIT601',
30-
'P601', 'P602', 'P603']
25+
label_col: "label"
26+
feature_cols: ['feature_1', 'feature_2', 'feature_3', 'feature_24']
3127

3228
model:
3329
name: TimesNet_AD

‎configs/anomaly/autoencoder_anomaly.yaml

+8-6
Original file line numberDiff line numberDiff line change
@@ -7,21 +7,23 @@ epoch: 1
77
training: True
88
anomaly_ratio: 1
99
task: anomaly
10+
to_static_train: False
11+
use_amp: False
12+
amp_level: O2
1013

1114

1215
dataset:
1316
name: TSADDataset
14-
dataset_root: ../data/
15-
train_path: ../data/MSL_train.csv
16-
val_path: ../data/MSL_test.csv
17+
dataset_root: ./dataset/
18+
train_path: ./dataset/ts_anomaly_examples/train.csv
19+
val_path: ./dataset/ts_anomaly_examples/test.csv
1720
scale: True
1821
time_feat: False
1922

2023
info_params:
2124
freq: 1
22-
label_col: label
23-
feature_cols: ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '10', '11', '12', '13', '14', '15', '16', '17', '18', '19', '20', '21', '22', '23', '24', '25', '26', '27', '28', '29', '30', '31', '32', '33', '34', '35', '36', '37', '38', '39', '40', '41', '42', '43', '44', '45', '46', '47', '48', '49', '50', '51', '52', '53', '54']
24-
25+
label_col: "label"
26+
feature_cols: ['feature_1', 'feature_2', 'feature_3', 'feature_24']
2527

2628
model:
2729
name: AutoEncoder

‎configs/anomaly/dlinear_anomaly.yaml

+10-6
Original file line numberDiff line numberDiff line change
@@ -7,21 +7,25 @@ epoch: 1
77
training: True
88
anomaly_ratio: 1
99
task: anomaly
10+
to_static_train: False
11+
use_amp: False
12+
amp_level: O2
13+
1014

1115

1216
dataset:
1317
name: TSADDataset
14-
dataset_root: ../data/
15-
train_path: ../data/PMS_train.csv
16-
val_path: ../data/PMS_test.csv
18+
dataset_root: ./dataset/
19+
train_path: ./dataset/ts_anomaly_examples/train.csv
20+
val_path: ./dataset/ts_anomaly_examples/test.csv
1721
scale: True
1822
time_feat: False
1923

2024
info_params:
2125
freq: 1
22-
time_col: timestamp
23-
feature_cols: ['feature_0', 'feature_1', 'feature_2', 'feature_3', 'feature_4', 'feature_5', 'feature_6', 'feature_7', 'feature_8', 'feature_9', 'feature_10', 'feature_11', 'feature_12', 'feature_13', 'feature_14', 'feature_15', 'feature_16', 'feature_17', 'feature_18', 'feature_19', 'feature_20', 'feature_21', 'feature_22', 'feature_23', 'feature_24']
24-
label_col: label
26+
label_col: "label"
27+
feature_cols: ['feature_1', 'feature_2', 'feature_3', 'feature_24']
28+
2529

2630
model:
2731
name: DLinear_AD

‎configs/anomaly/nlinear_anomaly.yaml

+9-5
Original file line numberDiff line numberDiff line change
@@ -7,20 +7,24 @@ epoch: 1
77
training: True
88
anomaly_ratio: 1
99
task: anomaly
10+
to_static_train: False
11+
use_amp: False
12+
amp_level: O2
13+
1014

1115

1216
dataset:
1317
name: TSADDataset
14-
dataset_root: ../data/
15-
train_path: ../data/MSL_train.csv
16-
val_path: ../data/MSL_test.csv
18+
dataset_root: ./dataset/
19+
train_path: ./dataset/ts_anomaly_examples/train.csv
20+
val_path: ./dataset/ts_anomaly_examples/test.csv
1721
scale: True
1822
time_feat: False
1923

2024
info_params:
2125
freq: 1
22-
label_col: label
23-
feature_cols: ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '10', '11', '12', '13', '14', '15', '16', '17', '18', '19', '20', '21', '22', '23', '24', '25', '26', '27', '28', '29', '30', '31', '32', '33', '34', '35', '36', '37', '38', '39', '40', '41', '42', '43', '44', '45', '46', '47', '48', '49', '50', '51', '52', '53', '54']
26+
label_col: "label"
27+
feature_cols: ['feature_1', 'feature_2', 'feature_3', 'feature_24']
2428

2529
# freq: 1
2630
# label_col: "Normal/Attack"

‎configs/anomaly/non_stationary_ad.yaml

+10-15
Original file line numberDiff line numberDiff line change
@@ -7,30 +7,25 @@ epoch: 1
77
training: True
88
anomaly_ratio: 1
99
task: anomaly
10+
to_static_train: False
11+
use_amp: False
12+
amp_level: O2
13+
1014

1115

1216
dataset:
1317
name: TSADDataset
14-
dataset_root: ../data/
15-
train_path: ../data/MSL_train.csv
16-
val_path: ../data/MSL_test.csv
18+
dataset_root: ./dataset/
19+
train_path: ./dataset/ts_anomaly_examples/train.csv
20+
val_path: ./dataset/ts_anomaly_examples/test.csv
1721
scale: True
1822
time_feat: False
1923

24+
2025
info_params:
21-
# freq: 1
22-
# label_col: "Normal/Attack"
23-
# feature_cols: ['FIT101', 'LIT101', ' MV101', 'P101', 'P102', ' AIT201', 'AIT202',
24-
# 'AIT203', 'FIT201', ' MV201', ' P201', ' P202', 'P203', ' P204',
25-
# 'P205', 'P206', 'DPIT301', 'FIT301', 'LIT301', 'MV301', 'MV302',
26-
# ' MV303', 'MV304', 'P301', 'P302', 'AIT401', 'AIT402', 'FIT401',
27-
# 'LIT401', 'P401', 'P402', 'P403', 'P404', 'UV401', 'AIT501',
28-
# 'AIT502', 'AIT503', 'AIT504', 'FIT501', 'FIT502', 'FIT503',
29-
# 'FIT504', 'P501', 'P502', 'PIT501', 'PIT502', 'PIT503', 'FIT601',
30-
# 'P601', 'P602', 'P603']
3126
freq: 1
32-
label_col: label
33-
feature_cols: ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '10', '11', '12', '13', '14', '15', '16', '17', '18', '19', '20', '21', '22', '23', '24', '25', '26', '27', '28', '29', '30', '31', '32', '33', '34', '35', '36', '37', '38', '39', '40', '41', '42', '43', '44', '45', '46', '47', '48', '49', '50', '51', '52', '53', '54']
27+
label_col: "label"
28+
feature_cols: ['feature_1', 'feature_2', 'feature_3', 'feature_24']
3429

3530

3631
model:

‎configs/anomaly/patchtst_anomaly.yaml

+8-17
Original file line numberDiff line numberDiff line change
@@ -7,32 +7,23 @@ epoch: 1
77
training: True
88
anomaly_ratio: 1
99
task: anomaly
10+
to_static_train: False
11+
use_amp: False
12+
amp_level: O2
1013

1114

1215
dataset:
1316
name: TSADDataset
14-
dataset_root: ../data/
15-
train_path: ../data/MSL_train.csv
16-
val_path: ../data/MSL_test.csv
17+
dataset_root: ./dataset/
18+
train_path: ./dataset/ts_anomaly_examples/train.csv
19+
val_path: ./dataset/ts_anomaly_examples/test.csv
1720
scale: True
1821
time_feat: False
1922

2023
info_params:
21-
# freq: 1
22-
# label_col: "Normal/Attack"
23-
# feature_cols: ['FIT101', 'LIT101', ' MV101', 'P101', 'P102', ' AIT201', 'AIT202',
24-
# 'AIT203', 'FIT201', ' MV201', ' P201', ' P202', 'P203', ' P204',
25-
# 'P205', 'P206', 'DPIT301', 'FIT301', 'LIT301', 'MV301', 'MV302',
26-
# ' MV303', 'MV304', 'P301', 'P302', 'AIT401', 'AIT402', 'FIT401',
27-
# 'LIT401', 'P401', 'P402', 'P403', 'P404', 'UV401', 'AIT501',
28-
# 'AIT502', 'AIT503', 'AIT504', 'FIT501', 'FIT502', 'FIT503',
29-
# 'FIT504', 'P501', 'P502', 'PIT501', 'PIT502', 'PIT503', 'FIT601',
30-
# 'P601', 'P602', 'P603']
31-
3224
freq: 1
33-
label_col: label
34-
feature_cols: ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '10', '11', '12', '13', '14', '15', '16', '17', '18', '19', '20', '21', '22', '23', '24', '25', '26', '27', '28', '29', '30', '31', '32', '33', '34', '35', '36', '37', '38', '39', '40', '41', '42', '43', '44', '45', '46', '47', '48', '49', '50', '51', '52', '53', '54']
35-
25+
label_col: "label"
26+
feature_cols: ['feature_1', 'feature_2', 'feature_3', 'feature_24']
3627

3728
model:
3829
name: PatchTST_AD

‎configs/anomaly/rlinear_anomaly.yaml

+9-16
Original file line numberDiff line numberDiff line change
@@ -7,31 +7,24 @@ epoch: 1
77
training: True
88
anomaly_ratio: 1
99
task: anomaly
10+
to_static_train: False
11+
use_amp: False
12+
amp_level: O2
13+
1014

1115

1216
dataset:
1317
name: TSADDataset
14-
dataset_root: ../data/
15-
train_path: ../data/MSL_train.csv
16-
val_path: ../data/MSL_test.csv
18+
dataset_root: ./dataset/
19+
train_path: ./dataset/ts_anomaly_examples/train.csv
20+
val_path: ./dataset/ts_anomaly_examples/test.csv
1721
scale: True
1822
time_feat: False
1923

2024
info_params:
21-
# freq: 1
22-
# label_col: "Normal/Attack"
23-
# feature_cols: ['FIT101', 'LIT101', ' MV101', 'P101', 'P102', ' AIT201', 'AIT202',
24-
# 'AIT203', 'FIT201', ' MV201', ' P201', ' P202', 'P203', ' P204',
25-
# 'P205', 'P206', 'DPIT301', 'FIT301', 'LIT301', 'MV301', 'MV302',
26-
# ' MV303', 'MV304', 'P301', 'P302', 'AIT401', 'AIT402', 'FIT401',
27-
# 'LIT401', 'P401', 'P402', 'P403', 'P404', 'UV401', 'AIT501',
28-
# 'AIT502', 'AIT503', 'AIT504', 'FIT501', 'FIT502', 'FIT503',
29-
# 'FIT504', 'P501', 'P502', 'PIT501', 'PIT502', 'PIT503', 'FIT601',
30-
# 'P601', 'P602', 'P603']
3125
freq: 1
32-
label_col: label
33-
feature_cols: ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '10', '11', '12', '13', '14', '15', '16', '17', '18', '19', '20', '21', '22', '23', '24', '25', '26', '27', '28', '29', '30', '31', '32', '33', '34', '35', '36', '37', '38', '39', '40', '41', '42', '43', '44', '45', '46', '47', '48', '49', '50', '51', '52', '53', '54']
34-
26+
label_col: "label"
27+
feature_cols: ['feature_1', 'feature_2', 'feature_3', 'feature_24']
3528

3629

3730
model:

‎configs/classification/timesnet_cls_base.yaml

+7-3
Original file line numberDiff line numberDiff line change
@@ -7,12 +7,16 @@ epoch: 30 # max_epochs
77
training: True #
88
eval_metrics: ['acc', ]
99
task: classification
10+
to_static_train: False
11+
use_amp: False
12+
amp_level: O2
13+
1014

1115
dataset:
1216
name: TSCLSDataset
13-
dataset_root: ./data/
14-
train_path: ./data/UWaveGestureLibrary_TRAIN.csv
15-
val_path: ./data/UWaveGestureLibrary_TEST.csv
17+
dataset_root: dataset/ts_classify_examples
18+
train_path: dataset/ts_classify_examples/train.csv
19+
val_path: dataset/ts_classify_examples/val.csv
1620
scale: True
1721
time_feat: False
1822

‎configs/longterm_forecast/DLinear_base.yaml

+6-3
Original file line numberDiff line numberDiff line change
@@ -4,12 +4,15 @@ predict_len: 336
44
sampling_stride: 1
55
do_eval: True
66
epoch: 10
7+
to_static_train: False
8+
use_amp: False
9+
amp_level: O2
710

811
dataset:
912
name: TSDataset
10-
dataset_root: /data/
11-
train_path: /data/train.csv
12-
val_path: /data/val.csv
13+
dataset_root: ts_dataset_examples
14+
train_path: ts_dataset_examples/train.csv
15+
val_path: ts_dataset_examples/val.csv
1316
time_feat: False
1417
scale: True
1518

‎configs/longterm_forecast/NLinear_base.yaml

+6-3
Original file line numberDiff line numberDiff line change
@@ -4,13 +4,16 @@ predict_len: 336
44
sampling_stride: 1
55
do_eval: True
66
epoch: 5
7+
to_static_train: False
8+
use_amp: False
9+
amp_level: O2
710

811

912
dataset:
1013
name: TSDataset
11-
dataset_root: /data/
12-
train_path: /data/train.csv
13-
val_path: /data/val.csv
14+
dataset_root: ts_dataset_examples
15+
train_path: ts_dataset_examples/train.csv
16+
val_path: ts_dataset_examples/val.csv
1417
time_feat: False
1518
scale: True
1619

‎configs/longterm_forecast/Nonstationary_base.yaml

+7-4
Original file line numberDiff line numberDiff line change
@@ -3,14 +3,17 @@ seq_len: 96
33
predict_len: 96
44
do_eval: True
55
sampling_stride: 1
6-
epoch: 5
6+
epoch: 2
7+
to_static_train: False
8+
use_amp: False
9+
amp_level: O2
710

811

912
dataset:
1013
name: TSDataset
11-
dataset_root: /data/
12-
train_path: /data/train.csv
13-
val_path: /data/val.csv
14+
dataset_root: ts_dataset_examples
15+
train_path: ts_dataset_examples/train.csv
16+
val_path: ts_dataset_examples/val.csv
1417
time_feat: True
1518
scale: True
1619

‎configs/longterm_forecast/PatchTST_base.yaml

+7-4
Original file line numberDiff line numberDiff line change
@@ -3,13 +3,16 @@ seq_len: 96
33
predict_len: 96
44
do_eval: True
55
sampling_stride: 1
6-
epoch: 5
6+
epoch: 2
7+
to_static_train: False
8+
use_amp: False
9+
amp_level: O2
710

811
dataset:
912
name: TSDataset
10-
dataset_root: /data/
11-
train_path: /data/train.csv
12-
val_path: /data/val.csv
13+
dataset_root: ts_dataset_examples
14+
train_path: ts_dataset_examples/train.csv
15+
val_path: ts_dataset_examples/val.csv
1316
time_feat: False
1417
scale: True
1518

‎configs/longterm_forecast/RLinear_base.yaml

+7-5
Original file line numberDiff line numberDiff line change
@@ -3,14 +3,16 @@ seq_len: 96
33
predict_len: 336
44
do_eval: True
55
sampling_stride: 1
6-
epoch: 10
7-
6+
epoch: 2
7+
to_static_train: False
8+
use_amp: False
9+
amp_level: O2
810

911
dataset:
1012
name: TSDataset
11-
dataset_root: /data/
12-
train_path: /data/train.csv
13-
val_path: /data/val.csv
13+
dataset_root: ts_dataset_examples
14+
train_path: ts_dataset_examples/train.csv
15+
val_path: ts_dataset_examples/val.csv
1416
time_feat: False
1517
scale: True
1618

‎configs/longterm_forecast/TiDE_base.yaml

+6-3
Original file line numberDiff line numberDiff line change
@@ -4,13 +4,16 @@ predict_len: 96
44
do_eval: True
55
sampling_stride: 1
66
epoch: 2
7+
to_static_train: False
8+
use_amp: False
9+
amp_level: O2
710

811

912
dataset:
1013
name: TSDataset
11-
dataset_root: /data/
12-
train_path: /data/train.csv
13-
val_path: /data/val.csv
14+
dataset_root: ts_dataset_examples
15+
train_path: ts_dataset_examples/train.csv
16+
val_path: ts_dataset_examples/val.csv
1417
time_feat: True
1518
use_holiday: True
1619
scale: True

0 commit comments

Comments
 (0)