| { | |
| "pipe-parallel-size": 1, | |
| "model-parallel-size": 1, | |
| "num-layers": 24, | |
| "hidden-size": 2048, | |
| "num-attention-heads": 16, | |
| "seq-length": 2048, | |
| "max-position-embeddings": 2048, | |
| "norm": "layernorm", | |
| "pos-emb": "rotary", | |
| "no-weight-tying": true, | |
| "gpt_j_residual": false, | |
| "output_layer_parallelism": "column", | |
| "attention_config": [[["flash"], 24]], | |
| "scaled-upper-triang-masked-softmax-fusion": false, | |
| "bias-gelu-fusion": false, | |
| "init_method": "small_init", | |
| "output_layer_init_method": "wang_init", | |
| "optimizer": { | |
| "type": "Adam", | |
| "params": { | |
| "lr": 0.0002, | |
| "betas": [0.9, 0.95], | |
| "eps": 1.0e-8 | |
| } | |
| }, | |
| "min_lr": 0.00002, | |
| "zero_optimization": { | |
| "stage": 1, | |
| "allgather_partitions": true, | |
| "allgather_bucket_size": 500000000, | |
| "overlap_comm": true, | |
| "reduce_scatter": true, | |
| "reduce_bucket_size": 500000000, | |
| "contiguous_gradients": true | |
| }, | |
| "train_micro_batch_size_per_gpu": 1, | |
| "autotuning": { | |
| "enabled": true, | |
| "arg_mappings": { | |
| "train_micro_batch_size_per_gpu": "--train_micro_batch_size_per_gpu", | |
| "gradient_accumulation_steps ": "--gradient_accumulation_steps" | |
| } | |
| }, | |
| "data-impl": "mmap", | |
| "checkpoint-activations": false, | |
| "checkpoint-num-layers": 1, | |
| "partition-activations": true, | |
| "synchronize-each-layer": true, | |
| "gradient_clipping": 1.0, | |
| "weight-decay": 0.1, | |
| "hidden-dropout": 0, | |
| "attention-dropout": 0, | |
| "fp16": { | |
| "fp16": true, | |
| "enabled": true, | |
| "loss_scale": 0, | |
| "loss_scale_window": 1000, | |
| "hysteresis": 2, | |
| "min_loss_scale": 1 | |
| }, | |
| "train-iters": 320000, | |
| "lr-decay-iters": 320000, | |
| "distributed-backend": "nccl", | |
| "lr-decay-style": "cosine", | |
| "warmup": 0.01, | |
| "checkpoint-factor": 10000, | |
| "eval-interval": 1000, | |
| "eval-iters": 10, | |
| "launcher": "slurm", | |
| "deepspeed_slurm": true, | |
| "no_ssh_check": true, | |
| "log-interval": 10, | |
| "steps_per_print": 10, | |
| "keep-last-n-checkpoints": 1, | |
| "wall_clock_breakdown": true | |
| } | |