add env var for deterministic training

open-mmlab · Feb 7, 2023 · 7eaf5b7 · 7eaf5b7
1 parent f46a3d7
commit 7eaf5b7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.dev_scripts/test_benchmark.py b/.dev_scripts/test_benchmark.py
@@ -163,6 +163,7 @@ def create_test_job_batch(commands, model_info, args, port, script_name):
                   f'#SBATCH --ntasks=2\n'
                   f'#SBATCH --cpus-per-task=16\n\n'
                   f'export MASTER_PORT={port}\n'
+                  f'export CUBLAS_WORKSPACE_CONFIG=:4096:8\n'
                   f'{runner} -u {script_name} {config} {checkpoint} '
                   f'--work-dir={work_dir} '
                   f'--out={result_file} '