From 171e2fcde636bcb7e6c0073a9983ed5252f04753 Mon Sep 17 00:00:00 2001 From: One Date: Mon, 21 Jul 2025 18:40:40 +0800 Subject: Update --- evaluate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'evaluate.py') diff --git a/evaluate.py b/evaluate.py index 9bc6ba0..71ee753 100644 --- a/evaluate.py +++ b/evaluate.py @@ -39,7 +39,7 @@ def launch(): # Dataloader train_loader, train_metadata = create_dataloader(config, "train", test_set_mode=False, epochs_per_iter=1, global_batch_size=config.global_batch_size, rank=RANK, world_size=WORLD_SIZE) - eval_loader, eval_metadata = create_dataloader(config, "test", test_set_mode=True, epochs_per_iter=1, global_batch_size=config.global_batch_size, test_set_limit_examples=LIMIT_EXAMPLES, rank=RANK, world_size=WORLD_SIZE) + eval_loader, eval_metadata = create_dataloader(config, "test", test_set_mode=True, epochs_per_iter=1, global_batch_size=config.global_batch_size, rank=RANK, world_size=WORLD_SIZE) # Models train_state = init_train_state(config, train_metadata, world_size=WORLD_SIZE) -- cgit v1.2.3