Removed debug dump of universal checkpoints

bigscience-workshop · Feb 21, 2023 · e52bdab · e52bdab
1 parent 09a35f5
commit e52bdab
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/megatron/training.py b/megatron/training.py
@@ -460,9 +460,6 @@ def setup_model_and_optimizer(model_provider_func):
 
     else:
         args.iteration = 0
-
-    from .utils import dump_weights 
-    dump_weights(f'{args.universal_checkpoint=}', args.iteration, model, optimizer)
 
     # tp_rank = mpu.get_tensor_model_parallel_rank()
     # pp_rank = mpu.get_pipeline_model_parallel_rank()