diff options
| author | One <imone@tuta.io> | 2025-07-21 18:40:40 +0800 |
|---|---|---|
| committer | One <imone@tuta.io> | 2025-07-21 18:40:40 +0800 |
| commit | 171e2fcde636bcb7e6c0073a9983ed5252f04753 (patch) | |
| tree | d7844d28ad5f289c25a046e58ec9d20216cfba44 /evaluate.py | |
| parent | bd6222774edcec1608a6842d0b06a637a4acef59 (diff) | |
Update
Diffstat (limited to 'evaluate.py')
| -rw-r--r-- | evaluate.py | 2 |
1 files changed, 1 insertions, 1 deletions
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) |