From dcd4f99f9cb09cc9180fd0fb522c1272575bfde2 Mon Sep 17 00:00:00 2001 From: czzhangheng Date: Mon, 10 Mar 2025 19:05:26 +0800 Subject: [PATCH] fix PDG2SEQ config --- config/PDG2SEQ/PEMSD4.yaml | 1 + config/PDG2SEQ/PEMSD7.yaml | 4 ++++ config/PDG2SEQ/PEMSD8.yaml | 4 ++++ 3 files changed, 9 insertions(+) diff --git a/config/PDG2SEQ/PEMSD4.yaml b/config/PDG2SEQ/PEMSD4.yaml index 6ccd606..687054d 100644 --- a/config/PDG2SEQ/PEMSD4.yaml +++ b/config/PDG2SEQ/PEMSD4.yaml @@ -25,6 +25,7 @@ model: lr_decay_step: 1500 lr_decay_step1: 60,75,90,120 time_dim: 16 + train: loss_func: mae seed: 10 diff --git a/config/PDG2SEQ/PEMSD7.yaml b/config/PDG2SEQ/PEMSD7.yaml index 3592e80..1b21274 100644 --- a/config/PDG2SEQ/PEMSD7.yaml +++ b/config/PDG2SEQ/PEMSD7.yaml @@ -12,9 +12,11 @@ data: test_ratio: 0.2 tod: false val_ratio: 0.2 + log: log_step: 3000 plot: false + model: cheb_k: 2 embed_dim: 12 @@ -27,9 +29,11 @@ model: lr_decay_step: 12000 lr_decay_step1: 80,100,120 time_dim: 20 + test: mae_thresh: None mape_thresh: 0.0 + train: batch_size: 16 early_stop: true diff --git a/config/PDG2SEQ/PEMSD8.yaml b/config/PDG2SEQ/PEMSD8.yaml index e4c4b95..9c6693a 100644 --- a/config/PDG2SEQ/PEMSD8.yaml +++ b/config/PDG2SEQ/PEMSD8.yaml @@ -12,9 +12,11 @@ data: test_ratio: 0.2 tod: false val_ratio: 0.2 + log: log_step: 2000 plot: false + model: cheb_k: 2 embed_dim: 12 @@ -27,9 +29,11 @@ model: lr_decay_step: 2000 lr_decay_step1: 50,75 time_dim: 16 + test: mae_thresh: None mape_thresh: 0.001 + train: batch_size: 64 early_stop: true