Skip to content

Commit

Permalink
feat: update configs;
Browse files Browse the repository at this point in the history
  • Loading branch information
WenjieDu committed May 28, 2024
1 parent e58543b commit 28402ad
Show file tree
Hide file tree
Showing 17 changed files with 34 additions and 33 deletions.
Original file line number Diff line number Diff line change
@@ -1,14 +1,14 @@
{
"n_steps": {"_type":"choice","_value":[48]},
"n_steps": {"_type":"choice","_value":[24]},
"n_features": {"_type":"choice","_value":[862]},
"epochs": {"_type":"choice","_value":[100]},
"patience": {"_type":"choice","_value":[10]},
"n_layers": {"_type":"choice","_value":[1,2,3]},
"d_model": {"_type":"choice","_value":[64,128,256,512,1024]},
"d_ffn": {"_type":"choice","_value":[64,128,256,512,1024]},
"n_heads": {"_type":"choice","_value":[1,2,4,8]},
"factor": {"_type":"choice","_value":[1,3,5]},
"moving_avg_window_size": {"_type":"choice","_value":[6,12,24]},
"factor": {"_type":"choice","_value":[3]},
"moving_avg_window_size": {"_type":"choice","_value":[5,13,25]},
"dropout": {"_type":"choice","_value":[0,0.1,0.2,0.3,0.4,0.5]},
"lr":{"_type":"loguniform","_value":[0.00005,0.01]}
}
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
{
"n_steps": {"_type":"choice","_value":[48]},
"n_steps": {"_type":"choice","_value":[24]},
"n_features": {"_type":"choice","_value":[862]},
"epochs": {"_type":"choice","_value":[100]},
"patience": {"_type":"choice","_value":[10]},
Expand Down
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
{
"n_steps": {"_type":"choice","_value":[48]},
"n_steps": {"_type":"choice","_value":[24]},
"n_features": {"_type":"choice","_value":[862]},
"epochs": {"_type":"choice","_value":[100]},
"patience": {"_type":"choice","_value":[10]},
"moving_avg_window_size": {"_type":"choice","_value":[6,12,24]},
"moving_avg_window_size": {"_type":"choice","_value":[5,13,25]},
"d_model": {"_type":"choice","_value":[64,128,256,512,1024]},
"lr":{"_type":"loguniform","_value":[0.00005,0.01]}
}
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
{
"n_steps": {"_type":"choice","_value":[48]},
"n_steps": {"_type":"choice","_value":[24]},
"n_features": {"_type":"choice","_value":[862]},
"epochs": {"_type":"choice","_value":[100]},
"patience": {"_type":"choice","_value":[10]},
Expand Down
Original file line number Diff line number Diff line change
@@ -1,19 +1,19 @@
experimentName: DLinear hyper-param searching
experimentName: ETSformer hyper-param searching
authorName: WenjieDu
trialConcurrency: 1
trainingServicePlatform: local
searchSpacePath: DLinear_PeMS_tuning_space.json
searchSpacePath: ETSformer_PeMS_tuning_space.json
multiThread: true
useAnnotation: false
tuner:
builtinTunerName: Random

trial:
command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.DLinear --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.ETSformer --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
codeDir: .
gpuNum: 1

localConfig:
useActiveGpu: true
maxTrialNumPerGpu: 20
gpuIndices: 1
gpuIndices: 4
Original file line number Diff line number Diff line change
@@ -1,13 +1,13 @@
{
"n_steps": {"_type":"choice","_value":[48]},
"n_steps": {"_type":"choice","_value":[24]},
"n_features": {"_type":"choice","_value":[862]},
"epochs": {"_type":"choice","_value":[100]},
"patience": {"_type":"choice","_value":[10]},
"window_size": {"_type":"choice","_value": [[128],[256]]},
"multiscale": {"_type":"choice","_value":[1,2,4]},
"window_size": {"_type":"choice","_value": [[2]]},
"multiscale": {"_type":"choice","_value":[[1,2]]},
"modes1": {"_type":"choice","_value":[32,64,128]},
"dropout": {"_type":"choice","_value":[0.1,0.2,0.3,0.4,0.5]},
"model_type": {"_type":"choice","_value":[0,1,2]},
"mode_type": {"_type":"choice","_value":[0,1,2]},
"d_model": {"_type":"choice","_value":[64,128,256,512,1024]},
"lr":{"_type":"loguniform","_value":[0.00005,0.01]}
}
Original file line number Diff line number Diff line change
@@ -1,10 +1,10 @@
{
"n_steps": {"_type":"choice","_value":[48]},
"n_steps": {"_type":"choice","_value":[24]},
"n_features": {"_type":"choice","_value":[862]},
"epochs": {"_type":"choice","_value":[100]},
"patience": {"_type":"choice","_value":[10]},
"embed_size": {"_type":"choice","_value":[64,128,256]},
"hidden_size": {"_type":"choice","_value":[64,128,256,512,1024]},
"hidden_size": {"_type":"choice","_value":[64,128,256]},
"channel_independence": {"_type":"choice","_value":[true,false]},
"lr":{"_type":"loguniform","_value":[0.00005,0.01]}
}
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
{
"n_steps": {"_type":"choice","_value":[48]},
"n_steps": {"_type":"choice","_value":[24]},
"n_features": {"_type":"choice","_value":[862]},
"epochs": {"_type":"choice","_value":[100]},
"patience": {"_type":"choice","_value":[10]},
Expand Down
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
{
"n_steps": {"_type":"choice","_value":[48]},
"n_steps": {"_type":"choice","_value":[24]},
"n_features": {"_type":"choice","_value":[862]},
"epochs": {"_type":"choice","_value":[100]},
"patience": {"_type":"choice","_value":[10]},
Expand Down
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
{
"n_steps": {"_type":"choice","_value":[48]},
"n_steps": {"_type":"choice","_value":[24]},
"n_features": {"_type":"choice","_value":[862]},
"epochs": {"_type":"choice","_value":[100]},
"patience": {"_type":"choice","_value":[10]},
Expand Down
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
{
"n_steps": {"_type":"choice","_value":[48]},
"n_steps": {"_type":"choice","_value":[24]},
"n_features": {"_type":"choice","_value":[862]},
"epochs": {"_type":"choice","_value":[100]},
"patience": {"_type":"choice","_value":[10]},
Expand Down
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
{
"n_steps": {"_type":"choice","_value":[48]},
"n_steps": {"_type":"choice","_value":[24]},
"n_features": {"_type":"choice","_value":[862]},
"epochs": {"_type":"choice","_value":[100]},
"patience": {"_type":"choice","_value":[10]},
Expand Down
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
{
"n_steps": {"_type":"choice","_value":[48]},
"n_steps": {"_type":"choice","_value":[24]},
"n_features": {"_type":"choice","_value":[862]},
"epochs": {"_type":"choice","_value":[100]},
"patience": {"_type":"choice","_value":[10]},
Expand All @@ -12,5 +12,6 @@
"d_k": {"_type":"choice","_value":[32, 64, 128, 256]},
"d_v": {"_type":"choice","_value":[32, 64, 128, 256]},
"dropout": {"_type":"choice","_value":[0,0.1,0.2,0.3,0.4,0.5]},
"attn_dropout": {"_type":"choice","_value":[0,0.1,0.2,0.3,0.4,0.5]},
"lr":{"_type":"loguniform","_value":[0.00005,0.01]}
}
Original file line number Diff line number Diff line change
@@ -1,13 +1,13 @@
{
"n_steps": {"_type":"choice","_value":[48]},
"n_steps": {"_type":"choice","_value":[24]},
"n_features": {"_type":"choice","_value":[862]},
"epochs": {"_type":"choice","_value":[100]},
"patience": {"_type":"choice","_value":[10]},
"n_layers": {"_type":"choice","_value":[1,2,3]},
"d_model": {"_type":"choice","_value":[64, 128, 256, 512, 1024]},
"d_ffn": {"_type":"choice","_value":[64,128,256,512,1024]},
"n_heads": {"_type":"choice","_value":[1,2,4,8]},
"window_size": {"_type":"choice","_value":[[4,4,4],[5,5,5]]},
"window_size": {"_type":"choice","_value":[[4,4]]},
"inner_size": {"_type":"choice","_value":[3,5]},
"dropout": {"_type":"choice","_value":[0,0.1,0.2,0.3,0.4,0.5]},
"attn_dropout": {"_type":"choice","_value":[0,0.1,0.2,0.3,0.4,0.5]},
Expand Down
Original file line number Diff line number Diff line change
@@ -1,13 +1,13 @@
{
"n_steps": {"_type":"choice","_value":[48]},
"n_steps": {"_type":"choice","_value":[24]},
"n_features": {"_type":"choice","_value":[862]},
"epochs": {"_type":"choice","_value":[100]},
"patience": {"_type":"choice","_value":[10]},
"n_stacks": {"_type":"choice","_value":[1,2,3]},
"n_levels": {"_type":"choice","_value":[1,2,3]},
"n_groups": {"_type":"choice","_value":[1,2,3]},
"n_decoder_layers": {"_type":"choice","_value":[1,2,3]},
"d_hidden": {"_type":"choice","_value":[64, 128, 256, 512, 1024]},
"n_stacks": {"_type":"choice","_value":[1,2]},
"n_levels": {"_type":"choice","_value":[1,2]},
"n_groups": {"_type":"choice","_value":[1]},
"n_decoder_layers": {"_type":"choice","_value":[1,2]},
"d_hidden": {"_type":"choice","_value":[64, 128, 256, 512]},
"dropout": {"_type":"choice","_value":[0,0.1,0.2,0.3,0.4,0.5]},
"lr":{"_type":"loguniform","_value":[0.00005,0.01]}
}
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
{
"n_steps": {"_type":"choice","_value":[48]},
"n_steps": {"_type":"choice","_value":[24]},
"n_features": {"_type":"choice","_value":[862]},
"epochs": {"_type":"choice","_value":[100]},
"patience": {"_type":"choice","_value":[10]},
Expand Down
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
{
"n_steps": {"_type":"choice","_value":[48]},
"n_steps": {"_type":"choice","_value":[24]},
"n_features": {"_type":"choice","_value":[862]},
"epochs": {"_type":"choice","_value":[100]},
"patience": {"_type":"choice","_value":[10]},
Expand Down

0 comments on commit 28402ad

Please sign in to comment.