Skip to content

Commit 2e36d63

Browse files
author
Charles Li
committed
Using synthetic dataset_type to decrease the training time
1 parent 77ecd7c commit 2e36d63

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

tests/integration/gradient_accumulation_test.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,7 @@ def test_grad_accumulate_same_loss(self):
6565
get_test_config_path(),
6666
f"base_output_directory={self.base_output_directory}",
6767
f"dataset_path={self.dataset_path}",
68+
"dataset_type=synthetic",
6869
"gradient_clipping_threshold=0", # Ensures we are testing raw scales of gradients (clipping off)
6970
"enable_checkpointing=False",
7071
"enable_goodput_recording=False",

0 commit comments

Comments
 (0)