We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c43406b commit 2199157Copy full SHA for 2199157
MaxText/tests/integration_tests/train_tests.py
@@ -322,9 +322,9 @@ def test_gpu_cudnn_flash_jax(self):
322
cudnn_flash_jax = [ # tests base config on GPU with flash attention"""
323
None,
324
os.path.join(PKG_DIR, "configs", "base.yml"),
325
- rf"base_output_directory=gs://runner-maxtext-logs",
+ "base_output_directory=gs://runner-maxtext-logs",
326
"run_name=runner_test",
327
- r"dataset_path=gs://maxtext-dataset",
+ "dataset_path=gs://maxtext-dataset",
328
"steps=2",
329
"enable_checkpointing=False",
330
"enable_goodput_recording=False",
0 commit comments