{ "train_batch_size" : 256, "train_micro_batch_size_per_gpu": 16, "steps_per_print": 100, "zero_optimization": { "stage": 0 }, "bf16": { "enabled": true } }