hybrid_engine_config.json 397 B

12345678910111213141516171819
  1. {
  2. "train_batch_size" : 32,
  3. "train_micro_batch_size_per_gpu": 2,
  4. "steps_per_print": 10,
  5. "zero_optimization": {
  6. "stage": 0,
  7. "offload_param": {
  8. "device": "cpu"
  9. },
  10. "stage3_param_persistence_threshold": 0
  11. },
  12. "fp16":{
  13. "enabled": true,
  14. "loss_scale_window": 100
  15. },
  16. "gradient_clipping": 1.0,
  17. "prescale_gradients": false,
  18. "wall_clock_breakdown" : false
  19. }