Skip to content

Commit

Permalink
Format the code and increase support for the MPS backend.
Browse files Browse the repository at this point in the history
  • Loading branch information
Tianxiang Zhan committed May 14, 2024
1 parent 2178821 commit 604cd3b
Show file tree
Hide file tree
Showing 774 changed files with 65,245 additions and 995 deletions.
86 changes: 85 additions & 1 deletion configs/ablation/ECL_script/TEFN_ac_p192.json
View file
Original file line number Diff line number Diff line change
@@ -1 +1,85 @@
{"expand": 2, "d_conv": 4, "top_k": 5, "num_kernels": 6, "enc_in": 321, "dec_in": 321, "c_out": 321, "d_model": 256, "n_heads": 8, "e_layers": 2, "d_layers": 1, "d_ff": 512, "moving_avg": 25, "factor": 3, "distil": true, "dropout": 0.1, "embed": "timeF", "activation": "gelu", "output_attention": false, "channel_independence": 0, "decomp_method": "moving_avg", "use_norm": 1, "down_sampling_layers": 0, "down_sampling_window": 1, "down_sampling_method": null, "seg_len": 48, "num_workers": 10, "itr": 1, "train_epochs": 10, "batch_size": 32, "patience": 3, "learning_rate": 0.01, "des": "Exp", "loss": "MSE", "lradj": "type1", "use_amp": false, "use_gpu": true, "gpu": 0, "use_multi_gpu": false, "devices": "0,1,2,3", "p_hidden_dims": [128, 128], "p_hidden_layers": 2, "use_dtw": false, "augmentation_ratio": 0, "seed": 2, "jitter": false, "scaling": false, "permutation": false, "randompermutation": false, "magwarp": false, "timewarp": false, "windowslice": false, "windowwarp": false, "rotation": false, "spawner": false, "dtwwarp": false, "shapedtwwarp": false, "wdba": false, "discdtw": false, "discsdtw": false, "extra_tag": "", "task_name": "long_term_forecast", "is_training": 1, "model_id": "ECL_96_192", "model": "TEFN_ac", "data": "custom", "root_path": "./dataset/electricity/", "data_path": "electricity.csv", "features": "M", "target": "OT", "freq": "h", "checkpoints": "./out/checkpoints/", "seq_len": 96, "label_len": 48, "pred_len": 192, "seasonal_patterns": "Monthly", "inverse": false, "mask_rate": 0.25, "anomaly_ratio": 0.25, "conv_kernel": null}
{
"expand": 2,
"d_conv": 4,
"top_k": 5,
"num_kernels": 6,
"enc_in": 321,
"dec_in": 321,
"c_out": 321,
"d_model": 256,
"n_heads": 8,
"e_layers": 2,
"d_layers": 1,
"d_ff": 512,
"moving_avg": 25,
"factor": 3,
"distil": true,
"dropout": 0.1,
"embed": "timeF",
"activation": "gelu",
"output_attention": false,
"channel_independence": 0,
"decomp_method": "moving_avg",
"use_norm": 1,
"down_sampling_layers": 0,
"down_sampling_window": 1,
"down_sampling_method": null,
"seg_len": 48,
"num_workers": 10,
"itr": 1,
"train_epochs": 10,
"batch_size": 32,
"patience": 3,
"learning_rate": 0.01,
"des": "Exp",
"loss": "MSE",
"lradj": "type1",
"use_amp": false,
"use_gpu": true,
"gpu": 0,
"use_multi_gpu": false,
"devices": "0,1,2,3",
"p_hidden_dims": [
128,
128
],
"p_hidden_layers": 2,
"use_dtw": false,
"augmentation_ratio": 0,
"seed": 2,
"jitter": false,
"scaling": false,
"permutation": false,
"randompermutation": false,
"magwarp": false,
"timewarp": false,
"windowslice": false,
"windowwarp": false,
"rotation": false,
"spawner": false,
"dtwwarp": false,
"shapedtwwarp": false,
"wdba": false,
"discdtw": false,
"discsdtw": false,
"extra_tag": "",
"task_name": "long_term_forecast",
"is_training": 1,
"model_id": "ECL_96_192",
"model": "TEFN_ac",
"data": "custom",
"root_path": "./dataset/electricity/",
"data_path": "electricity.csv",
"features": "M",
"target": "OT",
"freq": "h",
"checkpoints": "./out/checkpoints/",
"seq_len": 96,
"label_len": 48,
"pred_len": 192,
"seasonal_patterns": "Monthly",
"inverse": false,
"mask_rate": 0.25,
"anomaly_ratio": 0.25,
"conv_kernel": null
}
86 changes: 85 additions & 1 deletion configs/ablation/ECL_script/TEFN_ac_p336.json
View file
Original file line number Diff line number Diff line change
@@ -1 +1,85 @@
{"expand": 2, "d_conv": 4, "top_k": 5, "num_kernels": 6, "enc_in": 321, "dec_in": 321, "c_out": 321, "d_model": 256, "n_heads": 8, "e_layers": 1, "d_layers": 1, "d_ff": 512, "moving_avg": 25, "factor": 3, "distil": true, "dropout": 0.1, "embed": "timeF", "activation": "gelu", "output_attention": false, "channel_independence": 0, "decomp_method": "moving_avg", "use_norm": 1, "down_sampling_layers": 0, "down_sampling_window": 1, "down_sampling_method": null, "seg_len": 48, "num_workers": 10, "itr": 1, "train_epochs": 10, "batch_size": 32, "patience": 3, "learning_rate": 0.05, "des": "Exp", "loss": "MSE", "lradj": "type1", "use_amp": false, "use_gpu": true, "gpu": 0, "use_multi_gpu": false, "devices": "0,1,2,3", "p_hidden_dims": [128, 128], "p_hidden_layers": 2, "use_dtw": false, "augmentation_ratio": 0, "seed": 2, "jitter": false, "scaling": false, "permutation": false, "randompermutation": false, "magwarp": false, "timewarp": false, "windowslice": false, "windowwarp": false, "rotation": false, "spawner": false, "dtwwarp": false, "shapedtwwarp": false, "wdba": false, "discdtw": false, "discsdtw": false, "extra_tag": "", "task_name": "long_term_forecast", "is_training": 1, "model_id": "ECL_96_336", "model": "TEFN_ac", "data": "custom", "root_path": "./dataset/electricity/", "data_path": "electricity.csv", "features": "M", "target": "OT", "freq": "h", "checkpoints": "./out/checkpoints/", "seq_len": 96, "label_len": 48, "pred_len": 336, "seasonal_patterns": "Monthly", "inverse": false, "mask_rate": 0.25, "anomaly_ratio": 0.25, "conv_kernel": null}
{
"expand": 2,
"d_conv": 4,
"top_k": 5,
"num_kernels": 6,
"enc_in": 321,
"dec_in": 321,
"c_out": 321,
"d_model": 256,
"n_heads": 8,
"e_layers": 1,
"d_layers": 1,
"d_ff": 512,
"moving_avg": 25,
"factor": 3,
"distil": true,
"dropout": 0.1,
"embed": "timeF",
"activation": "gelu",
"output_attention": false,
"channel_independence": 0,
"decomp_method": "moving_avg",
"use_norm": 1,
"down_sampling_layers": 0,
"down_sampling_window": 1,
"down_sampling_method": null,
"seg_len": 48,
"num_workers": 10,
"itr": 1,
"train_epochs": 10,
"batch_size": 32,
"patience": 3,
"learning_rate": 0.05,
"des": "Exp",
"loss": "MSE",
"lradj": "type1",
"use_amp": false,
"use_gpu": true,
"gpu": 0,
"use_multi_gpu": false,
"devices": "0,1,2,3",
"p_hidden_dims": [
128,
128
],
"p_hidden_layers": 2,
"use_dtw": false,
"augmentation_ratio": 0,
"seed": 2,
"jitter": false,
"scaling": false,
"permutation": false,
"randompermutation": false,
"magwarp": false,
"timewarp": false,
"windowslice": false,
"windowwarp": false,
"rotation": false,
"spawner": false,
"dtwwarp": false,
"shapedtwwarp": false,
"wdba": false,
"discdtw": false,
"discsdtw": false,
"extra_tag": "",
"task_name": "long_term_forecast",
"is_training": 1,
"model_id": "ECL_96_336",
"model": "TEFN_ac",
"data": "custom",
"root_path": "./dataset/electricity/",
"data_path": "electricity.csv",
"features": "M",
"target": "OT",
"freq": "h",
"checkpoints": "./out/checkpoints/",
"seq_len": 96,
"label_len": 48,
"pred_len": 336,
"seasonal_patterns": "Monthly",
"inverse": false,
"mask_rate": 0.25,
"anomaly_ratio": 0.25,
"conv_kernel": null
}
86 changes: 85 additions & 1 deletion configs/ablation/ECL_script/TEFN_ac_p720.json
View file
Original file line number Diff line number Diff line change
@@ -1 +1,85 @@
{"expand": 2, "d_conv": 4, "top_k": 5, "num_kernels": 6, "enc_in": 321, "dec_in": 321, "c_out": 321, "d_model": 256, "n_heads": 8, "e_layers": 3, "d_layers": 1, "d_ff": 512, "moving_avg": 25, "factor": 3, "distil": true, "dropout": 0.1, "embed": "timeF", "activation": "gelu", "output_attention": false, "channel_independence": 0, "decomp_method": "moving_avg", "use_norm": 1, "down_sampling_layers": 0, "down_sampling_window": 1, "down_sampling_method": null, "seg_len": 48, "num_workers": 10, "itr": 1, "train_epochs": 10, "batch_size": 32, "patience": 3, "learning_rate": 0.01, "des": "Exp", "loss": "MSE", "lradj": "type1", "use_amp": false, "use_gpu": true, "gpu": 0, "use_multi_gpu": false, "devices": "0,1,2,3", "p_hidden_dims": [128, 128], "p_hidden_layers": 2, "use_dtw": false, "augmentation_ratio": 0, "seed": 2, "jitter": false, "scaling": false, "permutation": false, "randompermutation": false, "magwarp": false, "timewarp": false, "windowslice": false, "windowwarp": false, "rotation": false, "spawner": false, "dtwwarp": false, "shapedtwwarp": false, "wdba": false, "discdtw": false, "discsdtw": false, "extra_tag": "", "task_name": "long_term_forecast", "is_training": 1, "model_id": "ECL_96_720", "model": "TEFN_ac", "data": "custom", "root_path": "./dataset/electricity/", "data_path": "electricity.csv", "features": "M", "target": "OT", "freq": "h", "checkpoints": "./out/checkpoints/", "seq_len": 96, "label_len": 48, "pred_len": 720, "seasonal_patterns": "Monthly", "inverse": false, "mask_rate": 0.25, "anomaly_ratio": 0.25, "conv_kernel": null}
{
"expand": 2,
"d_conv": 4,
"top_k": 5,
"num_kernels": 6,
"enc_in": 321,
"dec_in": 321,
"c_out": 321,
"d_model": 256,
"n_heads": 8,
"e_layers": 3,
"d_layers": 1,
"d_ff": 512,
"moving_avg": 25,
"factor": 3,
"distil": true,
"dropout": 0.1,
"embed": "timeF",
"activation": "gelu",
"output_attention": false,
"channel_independence": 0,
"decomp_method": "moving_avg",
"use_norm": 1,
"down_sampling_layers": 0,
"down_sampling_window": 1,
"down_sampling_method": null,
"seg_len": 48,
"num_workers": 10,
"itr": 1,
"train_epochs": 10,
"batch_size": 32,
"patience": 3,
"learning_rate": 0.01,
"des": "Exp",
"loss": "MSE",
"lradj": "type1",
"use_amp": false,
"use_gpu": true,
"gpu": 0,
"use_multi_gpu": false,
"devices": "0,1,2,3",
"p_hidden_dims": [
128,
128
],
"p_hidden_layers": 2,
"use_dtw": false,
"augmentation_ratio": 0,
"seed": 2,
"jitter": false,
"scaling": false,
"permutation": false,
"randompermutation": false,
"magwarp": false,
"timewarp": false,
"windowslice": false,
"windowwarp": false,
"rotation": false,
"spawner": false,
"dtwwarp": false,
"shapedtwwarp": false,
"wdba": false,
"discdtw": false,
"discsdtw": false,
"extra_tag": "",
"task_name": "long_term_forecast",
"is_training": 1,
"model_id": "ECL_96_720",
"model": "TEFN_ac",
"data": "custom",
"root_path": "./dataset/electricity/",
"data_path": "electricity.csv",
"features": "M",
"target": "OT",
"freq": "h",
"checkpoints": "./out/checkpoints/",
"seq_len": 96,
"label_len": 48,
"pred_len": 720,
"seasonal_patterns": "Monthly",
"inverse": false,
"mask_rate": 0.25,
"anomaly_ratio": 0.25,
"conv_kernel": null
}
86 changes: 85 additions & 1 deletion configs/ablation/ECL_script/TEFN_ac_p96.json
View file
Original file line number Diff line number Diff line change
@@ -1 +1,85 @@
{"expand": 2, "d_conv": 4, "top_k": 5, "num_kernels": 6, "enc_in": 321, "dec_in": 321, "c_out": 321, "d_model": 256, "n_heads": 8, "e_layers": 0, "d_layers": 1, "d_ff": 512, "moving_avg": 25, "factor": 3, "distil": true, "dropout": 0.1, "embed": "timeF", "activation": "gelu", "output_attention": false, "channel_independence": 0, "decomp_method": "moving_avg", "use_norm": 1, "down_sampling_layers": 0, "down_sampling_window": 1, "down_sampling_method": null, "seg_len": 48, "num_workers": 10, "itr": 1, "train_epochs": 10, "batch_size": 32, "patience": 3, "learning_rate": 0.05, "des": "Exp", "loss": "MSE", "lradj": "type1", "use_amp": false, "use_gpu": true, "gpu": 0, "use_multi_gpu": false, "devices": "0,1,2,3", "p_hidden_dims": [128, 128], "p_hidden_layers": 2, "use_dtw": false, "augmentation_ratio": 0, "seed": 2, "jitter": false, "scaling": false, "permutation": false, "randompermutation": false, "magwarp": false, "timewarp": false, "windowslice": false, "windowwarp": false, "rotation": false, "spawner": false, "dtwwarp": false, "shapedtwwarp": false, "wdba": false, "discdtw": false, "discsdtw": false, "extra_tag": "", "task_name": "long_term_forecast", "is_training": 1, "model_id": "ECL_96_96", "model": "TEFN_ac", "data": "custom", "root_path": "./dataset/electricity/", "data_path": "electricity.csv", "features": "M", "target": "OT", "freq": "h", "checkpoints": "./out/checkpoints/", "seq_len": 96, "label_len": 48, "pred_len": 96, "seasonal_patterns": "Monthly", "inverse": false, "mask_rate": 0.25, "anomaly_ratio": 0.25, "conv_kernel": null}
{
"expand": 2,
"d_conv": 4,
"top_k": 5,
"num_kernels": 6,
"enc_in": 321,
"dec_in": 321,
"c_out": 321,
"d_model": 256,
"n_heads": 8,
"e_layers": 0,
"d_layers": 1,
"d_ff": 512,
"moving_avg": 25,
"factor": 3,
"distil": true,
"dropout": 0.1,
"embed": "timeF",
"activation": "gelu",
"output_attention": false,
"channel_independence": 0,
"decomp_method": "moving_avg",
"use_norm": 1,
"down_sampling_layers": 0,
"down_sampling_window": 1,
"down_sampling_method": null,
"seg_len": 48,
"num_workers": 10,
"itr": 1,
"train_epochs": 10,
"batch_size": 32,
"patience": 3,
"learning_rate": 0.05,
"des": "Exp",
"loss": "MSE",
"lradj": "type1",
"use_amp": false,
"use_gpu": true,
"gpu": 0,
"use_multi_gpu": false,
"devices": "0,1,2,3",
"p_hidden_dims": [
128,
128
],
"p_hidden_layers": 2,
"use_dtw": false,
"augmentation_ratio": 0,
"seed": 2,
"jitter": false,
"scaling": false,
"permutation": false,
"randompermutation": false,
"magwarp": false,
"timewarp": false,
"windowslice": false,
"windowwarp": false,
"rotation": false,
"spawner": false,
"dtwwarp": false,
"shapedtwwarp": false,
"wdba": false,
"discdtw": false,
"discsdtw": false,
"extra_tag": "",
"task_name": "long_term_forecast",
"is_training": 1,
"model_id": "ECL_96_96",
"model": "TEFN_ac",
"data": "custom",
"root_path": "./dataset/electricity/",
"data_path": "electricity.csv",
"features": "M",
"target": "OT",
"freq": "h",
"checkpoints": "./out/checkpoints/",
"seq_len": 96,
"label_len": 48,
"pred_len": 96,
"seasonal_patterns": "Monthly",
"inverse": false,
"mask_rate": 0.25,
"anomaly_ratio": 0.25,
"conv_kernel": null
}
Loading

0 comments on commit 604cd3b

Please sign in to comment.