Skip to content

Commit

Permalink
upload pre-trained models for all datasets
Browse files Browse the repository at this point in the history
  • Loading branch information
YunhaoZhang-Mars committed May 28, 2024
1 parent b40526a commit 6f1f262
Show file tree
Hide file tree
Showing 17 changed files with 81 additions and 2 deletions.
3 changes: 1 addition & 2 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -3,5 +3,4 @@ forecast_checkpoints/
detect_result/
forecast_results/
*.pyc
datasets/
pretrain-library/
datasets/
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
{
"data_dim": 321,
"patch_size": 12,
"d_model": 256,
"d_ff": 512,
"n_heads": 4,
"e_layers": 4,
"d_layers": 1,
"dropout": 0.0
}
Binary file not shown.
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
{
"data_dim": 7,
"patch_size": 12,
"d_model": 256,
"d_ff": 512,
"n_heads": 4,
"e_layers": 4,
"d_layers": 1,
"dropout": 0.0
}
Binary file not shown.
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
{
"data_dim": 9,
"patch_size": 10,
"d_model": 256,
"d_ff": 512,
"n_heads": 4,
"e_layers": 4,
"d_layers": 1,
"dropout": 0.0
}
Binary file not shown.
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
{
"data_dim": 25,
"patch_size": 10,
"d_model": 256,
"d_ff": 512,
"n_heads": 4,
"e_layers": 4,
"d_layers": 1,
"dropout": 0.0
}
Binary file not shown.
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
{
"data_dim": 38,
"patch_size": 10,
"d_model": 256,
"d_ff": 512,
"n_heads": 4,
"e_layers": 4,
"d_layers": 1,
"dropout": 0.0
}
Binary file not shown.
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
{
"data_dim": 51,
"patch_size": 10,
"d_model": 256,
"d_ff": 512,
"n_heads": 4,
"e_layers": 4,
"d_layers": 1,
"dropout": 0.0
}
Binary file not shown.
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
{
"data_dim": 862,
"patch_size": 12,
"d_model": 256,
"d_ff": 512,
"n_heads": 4,
"e_layers": 4,
"d_layers": 1,
"dropout": 0.0
}
Binary file not shown.
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
{
"data_dim": 21,
"patch_size": 12,
"d_model": 256,
"d_ff": 512,
"n_heads": 4,
"e_layers": 4,
"d_layers": 1,
"dropout": 0.0
}
Binary file not shown.

0 comments on commit 6f1f262

Please sign in to comment.