Skip to content

GitLab

Commit c64098ef authored by Lawrence McAfee's avatar Lawrence McAfee
Browse files

tiny change.

parent 5fb0ff9a
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
megatron/training.py
View file @ c64098ef
...@@ -457,7 +457,7 @@ def train_step(forward_step_func, data_iterator, ...@@ -457,7 +457,7 @@ def train_step(forward_step_func, data_iterator,
# <<< # <<<
# >>> # >>>
optimizer.debug_model(ITERATION, "after gather params.", 0) # optimizer.debug_model(ITERATION, "after gather params.", 0)
# <<< # <<<
if args.vision_pretraining and args.vision_pretraining_type == "dino": if args.vision_pretraining and args.vision_pretraining_type == "dino":
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment