From d2e1d5e64d6067c3009b380d276142441acc7db4 Mon Sep 17 00:00:00 2001 From: Mihir Patel Date: Mon, 23 Sep 2024 11:15:31 -0700 Subject: [PATCH] Fix CPU dailies (#3628) --- tests/callbacks/callback_settings.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tests/callbacks/callback_settings.py b/tests/callbacks/callback_settings.py index 279f47a695..4b3ef478a2 100644 --- a/tests/callbacks/callback_settings.py +++ b/tests/callbacks/callback_settings.py @@ -146,6 +146,11 @@ NeptuneLogger: { 'mode': 'debug', }, + WandBLogger: { + 'init_kwargs': { + 'mode': 'offline', + }, + }, composer.profiler.Profiler: { 'trace_handlers': [MagicMock()], 'schedule': composer.profiler.cyclic_schedule(),