From b557f087ac9a90019a8cf7b4f016285eb1bf936a Mon Sep 17 00:00:00 2001 From: hariharandev1 Date: Sat, 5 Oct 2024 01:56:25 -0700 Subject: [PATCH] fixed calculation --- tests/dlio_benchmark_test.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tests/dlio_benchmark_test.py b/tests/dlio_benchmark_test.py index 834d7ab8..f9a63d0f 100644 --- a/tests/dlio_benchmark_test.py +++ b/tests/dlio_benchmark_test.py @@ -273,7 +273,6 @@ def test_checkpoint_epoch(framework, model_size, optimizers, num_layers, layer_p num_optimizer_files = 1 num_layer_files = num_layers files_per_checkpoint = (num_model_files + num_optimizer_files + num_layer_files) * nranks - comm.Barrier() if framework == "tensorflow": file_per_ckp = 2 num_check_files = epochs / epoch_per_ckp * files_per_checkpoint * file_per_ckp + 1