test

ecmwf · Aug 2, 2024 · 4b43694 · 4b43694
1 parent 9883299
commit 4b43694
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/diagnostics/test_checkpoint.py b/tests/diagnostics/test_checkpoint.py
@@ -17,7 +17,7 @@
 from anemoi.training.utils.jsonify import map_config_to_primitives
 
 os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
-os.environ["PYTORCH_MPS_HIGH_WATERMARK_RATIO"] = "0"
+os.environ["PYTORCH_MPS_HIGH_WATERMARK_RATIO"] = "2.0"
 
 
 class DummyModel(torch.nn.Module):