Commits · ed90a1b21de6f0682e1cfb370434b5df4f6fbeaf · OpenDAS / Megatron-LM · GitLab

18 Mar, 2022 4 commits
- mem savings factor training; need to check loss. · ed90a1b2
  Lawrence McAfee authored Mar 18, 2022
  
  ed90a1b2
- added arg '--distrib-opt-comm-mem-savings'. · 06b9ebe0
  Lawrence McAfee authored Mar 18, 2022
  
  06b9ebe0
- after experiments, before new arg 'distrib-opt-comm-buf-1m'. · 34fc1e37
  Lawrence McAfee authored Mar 18, 2022
  
  34fc1e37
- split reduce_grads/gather_params using gbuf sub-views. · 12d91733
  Lawrence McAfee authored Mar 18, 2022
  
  12d91733
14 Mar, 2022 11 commits
- cleaned arguments.py. · e30ad67e
  Lawrence McAfee authored Mar 14, 2022
  
  e30ad67e
- cleaned distributed.py. · 14e60427
  Lawrence McAfee authored Mar 14, 2022
  
  14e60427
- cleaned distrib_optimizer.py. · 2c1660e7
  Lawrence McAfee authored Mar 14, 2022
  
  2c1660e7
- partially cleaned optimizer.py. · efa3cbcf
  Lawrence McAfee authored Mar 14, 2022
  
  efa3cbcf
- cleaned optimizer/__init__.py. · e6120623
  Lawrence McAfee authored Mar 14, 2022
  
  e6120623
- cleaned clip_grads.py. · a4afb428
  Lawrence McAfee authored Mar 14, 2022
  
  a4afb428
- cleaned training.py · 787882a0
  Lawrence McAfee authored Mar 14, 2022
  
  787882a0
- cleaned/commented checkpointing.py. · 0f2a9f19
  Lawrence McAfee authored Mar 14, 2022
  
  0f2a9f19
- more specific formatting of model/optim checkpoint paths. · 9c86abd9
  Lawrence McAfee authored Mar 14, 2022
  
  9c86abd9
- optimizer saves list(group), not list(param). · af2b136f
  Lawrence McAfee authored Mar 14, 2022
  
  af2b136f
- started updating checkpointing.py. · 37ca7859
  Lawrence McAfee authored Mar 14, 2022
  
  37ca7859
11 Mar, 2022 3 commits
- guard 'gather_params()' with 'if update_successful:' · f6811e28
  Lawrence McAfee authored Mar 11, 2022
  
  f6811e28
- tiny change. · c64098ef
  Lawrence McAfee authored Mar 10, 2022
  
  c64098ef
- Merge branch 'main' into lmcafee/distrib-opt · 5fb0ff9a
  Lawrence McAfee authored Mar 10, 2022
  
  5fb0ff9a
10 Mar, 2022 2 commits
- small fixes. · 862d70fc
  Lawrence McAfee authored Mar 10, 2022
  
  862d70fc
- updated args for allreduce_embeddings · 08ee8ea2
  Lawrence McAfee authored Mar 10, 2022
  
  08ee8ea2
09 Mar, 2022 14 commits
- updated FP32Optimizer for latest changes. · c88bc979
  Lawrence McAfee authored Mar 09, 2022
  
  c88bc979
- removed old (pre-gbuf-padding) get_model_grad_buffer_dp_views(). · 7ac342b7
  Lawrence McAfee authored Mar 09, 2022
  
  7ac342b7
- more cleanup · 0528bf77
  Lawrence McAfee authored Mar 09, 2022
  
  0528bf77
- removed faulty (already commented out) gather_model_params param iterator. · 94a90215
  Lawrence McAfee authored Mar 09, 2022
  
  94a90215
- renamed reduce_grads/gather_params -> reduce_model_grads/gather_model_params · 64b9d28c
  Lawrence McAfee authored Mar 09, 2022
  
  64b9d28c
- passing args, timers from train_step to optimizer methods. · d58d1762
  Lawrence McAfee authored Mar 09, 2022
  
  d58d1762
- consolidated reduce_grad's sub-methods (i.e., allreduce_embedding_grads) · 45b364b1
  Lawrence McAfee authored Mar 09, 2022
  
  45b364b1
- finished cleaning clip_grads.py · a9b1fc0a
  Lawrence McAfee authored Mar 09, 2022
  
  a9b1fc0a
- passing 'model_parallel_group' to clip_grads, count_zeros · 9546d8f0
  Lawrence McAfee authored Mar 09, 2022
  
  9546d8f0
- moved 'reduce_grads()' to MegatronOptimizer. · e46230dc
  Lawrence McAfee authored Mar 09, 2022
  
  e46230dc
- encapsulated 'get_model_parallel_group()'; renamed BaseFloat16Optimizer -> MixedPrecisionOptimizer · 772a4a2d
  Lawrence McAfee authored Mar 09, 2022
  
  772a4a2d
- removed clip_grad_norm shortcut. · 6e888151
  Lawrence McAfee authored Mar 09, 2022
  
  6e888151
- moved distrib opt to own file. · 28b63ec7
  Lawrence McAfee authored Mar 09, 2022
  
  28b63ec7
- padded ddp's grad_buffer to multiple of data parallel world size · 55695f81
  Lawrence McAfee authored Mar 09, 2022
  
  55695f81
07 Mar, 2022 1 commit
- deleted old distributed_fused_adam · 4f2356dc
  Lawrence McAfee authored Mar 07, 2022
  
  4f2356dc
02 Mar, 2022 3 commits
- working w/ mixed parallelism [ dp, tp, pp ]. · e5bda3c9
  Lawrence McAfee authored Mar 02, 2022
  
  e5bda3c9
- copied t5 embed sync to distrib opt · 2f99b4f4
  Lawrence McAfee authored Mar 02, 2022
  
  2f99b4f4
- fixed gather params; now copying all params, not just local dp subset. · f8fd34e3
  Lawrence McAfee authored Mar 02, 2022
  
  f8fd34e3
01 Mar, 2022 2 commits
- fixed 'count zeros' for distrib opt · cbcd5579
  Lawrence McAfee authored Mar 01, 2022
  
  cbcd5579
- more debugging; some cleanup · 67e23459
  Lawrence McAfee authored Feb 28, 2022
  
  67e23459