clean spaces

NVIDIA · Oct 28, 2024 · 54c9bdd · 54c9bdd
1 parent 9e6458d
commit 54c9bdd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/megatron/training/training.py b/megatron/training/training.py
@@ -301,7 +301,7 @@ def pretrain(
     model, optimizer, opt_param_scheduler = setup_model_and_optimizer(
         model_provider, model_type, 
         scale_lr_cond=(lambda name, param: args.scale_lr_layer in name) if args.scale_lr_layer else None,
-        lr_mult= args.lr_multiplier,
+        lr_mult=args.lr_multiplier,
         checkpointing_context=checkpointing_context)
 
     timers('model-and-optimizer-setup').stop()

diff --git a/tasks/vision/finetune_utils.py b/tasks/vision/finetune_utils.py
@@ -225,7 +225,7 @@ def finetune(
             model_provider,
             model_type,
             scale_lr_cond=(lambda name, param: args.scale_lr_layer in name) if args.scale_lr_layer else None,
-            lr_mult= args.lr_multiplier)
+            lr_mult=args.lr_multiplier)
     timers("model and optimizer").stop()
 
     # If pretrained checkpoint is provided and we have not trained for