reduce splitsize

mlcommons · Sep 26, 2023 · e692c07 · e692c07
1 parent ac17834
commit e692c07
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/submission_runner.py b/submission_runner.py
@@ -15,6 +15,8 @@
 """
 import os
 # os.environ["CUDA_VISIBLE_DEVICES"]='0'
+os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:256"
+
 
 import datetime
 import gc