diff options
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) |