From 0c86f7113b709df7ce3bc995ba34457c6970cfed Mon Sep 17 00:00:00 2001 From: Huihuo Zheng Date: Thu, 6 Feb 2025 10:19:46 -0600 Subject: [PATCH] fixing test --- tests/dlio_benchmark_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/dlio_benchmark_test.py b/tests/dlio_benchmark_test.py index 5807de9e..ba7d49dc 100644 --- a/tests/dlio_benchmark_test.py +++ b/tests/dlio_benchmark_test.py @@ -262,7 +262,7 @@ def test_checkpoint_epoch(framework, model_size, optimizers, num_layers, layer_p comm.Barrier() benchmark = run_benchmark(cfg) output = pathlib.Path("./checkpoints") - load_bin = list(output.glob("*")) + load_bin = list(output.glob("*/*.pt")) n = 0 if len(layer_params) > 0: n = num_layers