Skip to content

Commit

Permalink
remove a spurious print that is making the table a bit uglier
Browse files Browse the repository at this point in the history
  • Loading branch information
karpathy committed Jun 21, 2024
1 parent c6e21c7 commit 72a2158
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion train_gpt2.cu
Original file line number Diff line number Diff line change
Expand Up @@ -97,7 +97,6 @@ void set_zero_configs(MultiGpuConfig* multi_gpu_config, int zero_stage, size_t t
multi_gpu_config->zero_stage = 0;
}
else {
printf0("| Zero Stage1 is enabled |\n");
multi_gpu_config->zero_stage = 1;
multi_gpu_config->shard_num_parameters = total_parameters / multi_gpu_config->num_processes;
}
Expand Down

0 comments on commit 72a2158

Please sign in to comment.