We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0cf6e23 commit ca86599Copy full SHA for ca86599
dags/orbax/util/test_config_util.py
@@ -189,7 +189,7 @@ def generate_workload_command(
189
command = (
190
f"export TPU_PREMAPPED_BUFFER_SIZE={tpu_premmapped_size} && "
191
f"export TPU_PREMAPPED_BUFFER_TRANSFER_THRESHOLD_BYTES={tpu_premmapped_size} && "
192
- "python3 -m MaxText.train MaxText/configs/base.yml "
+ "python3 -m MaxText.train maxtext/configs/base.yml "
193
"remat_policy=full "
194
"global_parameter_scale=1 "
195
f"base_output_directory={posixpath.join(self.base_dir, out_folder)} "
0 commit comments