From 1e710860d7d9c927fbb28e4601a2dddfaea92166 Mon Sep 17 00:00:00 2001 From: Gert Mertes Date: Fri, 2 Aug 2024 11:55:41 +0000 Subject: [PATCH] test --- tests/diagnostics/test_checkpoint.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/diagnostics/test_checkpoint.py b/tests/diagnostics/test_checkpoint.py index 030fe7d1..ed7d6259 100644 --- a/tests/diagnostics/test_checkpoint.py +++ b/tests/diagnostics/test_checkpoint.py @@ -15,7 +15,7 @@ from anemoi.training.diagnostics.callbacks import AnemoiCheckpoint from anemoi.training.utils.jsonify import map_config_to_primitives -torch.mps.set_per_process_memory_fraction(0.0) +torch.mps.set_per_process_memory_fraction(0.7) class DummyModel(torch.nn.Module):