Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Multi spans w.o executor #415

Merged
merged 12 commits into from
May 21, 2024
Merged

Multi spans w.o executor #415

merged 12 commits into from
May 21, 2024

Conversation

Kazeya27
Copy link
Collaborator

  • 数据集配置文件(TrafficStatePointDataset.json)

{ "batch_size": 32, "cache_dataset": true, "num_workers": 0, "pad_with_last_sample": true, "train_rate": 0.6, "eval_rate": 0.2, "scaler": "standard", "load_external": true, "normal_external": false, "ext_scaler": "none", "input_window": 12, "output_window": 12, "add_time_in_day": true, "add_day_in_week": true }

  • 执行器配置文件(TrafficStateExecutor)

{ "gpu": true, "gpu_id": 0, "max_epoch": 100, "train_loss": "masked_mae", "epoch": 0, "learner": "adam", "learning_rate": 0.01, "weight_decay": 0, "lr_epsilon": 1e-8, "lr_beta1": 0.9, "lr_beta2": 0.999, "lr_alpha": 0.99, "lr_momentum": 0, "lr_decay": false, "lr_scheduler": "multisteplr", "lr_decay_ratio": 0.1, "steps": [5, 20, 40, 70], "step_size": 10, "lr_T_max": 30, "lr_eta_min": 0, "lr_patience": 10, "lr_threshold": 1e-4, "clip_grad_norm": false, "max_grad_norm": 1.0, "use_early_stop": false, "patience": 50, "log_level": "INFO", "log_every": 1, "saved_model": true, "load_best_epoch": true, "hyper_tune": false }

@Kazeya27 Kazeya27 requested a review from NickHan-cs May 14, 2024 04:36
@Kazeya27
Copy link
Collaborator Author

PEMSD8-flow LibCity MultiSPANS
MAE 15.04 15.04
RMSE 24.31 24.33

@Kazeya27 Kazeya27 merged commit c0e7dd0 into master May 21, 2024
@Kazeya27 Kazeya27 deleted the MultiSPANS-w.o_executor branch May 21, 2024 06:39
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants