ModuFlow/mask-ddpm
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

result

Browse Source
This commit is contained in:
root
2026-02-04 06:05:53 +08:00
parent 10c0721ee1
commit 8e8ddbb7db
68 changed files with 14864 additions and 2859 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
example/results/config_used.json
View File

@@ -7,7 +7,7 @@
"out_dir": "/root/autodl-tmp/mask-ddpm/mask-ddpm/example/results",
"device": "cuda",
"timesteps": 600,
"batch_size": 16,
"batch_size": 12,
"seq_len": 96,
"epochs": 10,
"max_batches": 4000,
@@ -39,10 +39,17 @@
"cont_target": "x0",
"cont_clamp_x0": 5.0,
"use_temporal_stage1": true,
"temporal_backbone": "transformer",
"temporal_hidden_dim": 256,
"temporal_num_layers": 1,
"temporal_dropout": 0.0,
"temporal_epochs": 2,
"temporal_pos_dim": 64,
"temporal_use_pos_embed": true,
"temporal_transformer_num_layers": 2,
"temporal_transformer_nhead": 4,
"temporal_transformer_ff_dim": 256,
"temporal_transformer_dropout": 0.1,
"temporal_epochs": 3,
"temporal_lr": 0.001,
"quantile_loss_weight": 0.2,
"quantile_points": [
@@ -56,7 +63,7 @@
"snr_gamma": 1.0,
"residual_stat_weight": 0.05,
"backbone_type": "transformer",
"transformer_num_layers": 2,
"transformer_num_layers": 3,
"transformer_nhead": 4,
"transformer_ff_dim": 512,
"transformer_dropout": 0.1,
@@ -99,5 +106,5 @@
"P2_HILout"
],
"sample_batch_size": 4,
"sample_seq_len": 128
"sample_seq_len": 96
}