xingjianleng commited on
Commit
016cf4a
·
verified ·
1 Parent(s): 806242e

Delete args.json

Browse files
Files changed (1) hide show
  1. args.json +0 -55
args.json DELETED
@@ -1,55 +0,0 @@
1
- {
2
- "output_dir": "exps_simple_e2e",
3
- "exp_name": "sit-xl-1-f16d64-e2e-novariational-logvar0.0",
4
- "logging_dir": "logs",
5
- "report_to": "wandb",
6
- "sampling_steps": 50000,
7
- "resume_step": 0,
8
- "force_val_bsz": false,
9
- "model": "SiT-XL/1",
10
- "num_classes": 1000,
11
- "qk_norm": false,
12
- "fused_attn": true,
13
- "bn_momentum": 0.1,
14
- "data_dir": "/scratch/xingjian.leng/data",
15
- "resolution": 256,
16
- "batch_size": 256,
17
- "allow_tf32": true,
18
- "mixed_precision": "bf16",
19
- "epochs": 1400,
20
- "max_train_steps": 400000,
21
- "checkpointing_steps": 50000,
22
- "gradient_accumulation_steps": 1,
23
- "learning_rate": 0.0001,
24
- "adam_beta1": 0.9,
25
- "adam_beta2": 0.999,
26
- "adam_weight_decay": 0.0,
27
- "adam_epsilon": 1e-08,
28
- "max_grad_norm": 1.0,
29
- "grad_check_threshold": 10.0,
30
- "compile": true,
31
- "seed": 0,
32
- "num_workers": 18,
33
- "path_type": "linear",
34
- "prediction": "v",
35
- "cfg_prob": 0.1,
36
- "weighting": "uniform",
37
- "vae": "f16d64",
38
- "vae_ckpt": null,
39
- "fixed_logvar": 0.0,
40
- "use_variational": false,
41
- "disc_pretrained_ckpt": null,
42
- "loss_cfg_path": "configs/l1_lpips_kl_gan.yaml",
43
- "vae_learning_rate": 0.0001,
44
- "vae_adam_beta1": 0.9,
45
- "vae_adam_beta2": 0.999,
46
- "vae_adam_weight_decay": 0.0,
47
- "vae_adam_epsilon": 1e-08,
48
- "vae_scheduler_type": "constant",
49
- "vae_warmup_steps": 0,
50
- "disc_learning_rate": 0.0001,
51
- "disc_adam_beta1": 0.9,
52
- "disc_adam_beta2": 0.999,
53
- "disc_adam_weight_decay": 0.0,
54
- "disc_adam_epsilon": 1e-08
55
- }