From 95df0904ae5d2b3aaa26b708e5067e9271624036 Mon Sep 17 00:00:00 2001 From: Richard Barnes Date: Tue, 5 Sep 2023 17:50:44 -0700 Subject: [PATCH] Del `(object)` from 10 inc pytorch/inplace_abn/scripts/imagenet/utils.py Summary: Python3 makes the use of `(object)` in class inheritance unnecessary. Let's modernize our code by eliminating this. Reviewed By: meyering Differential Revision: D48957980 fbshipit-source-id: ef91fea76e0d4647d1c447429b8c377e81e121ab --- opacus/utils/batch_memory_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/opacus/utils/batch_memory_manager.py b/opacus/utils/batch_memory_manager.py index 8f757e5d..c5d6dcc0 100644 --- a/opacus/utils/batch_memory_manager.py +++ b/opacus/utils/batch_memory_manager.py @@ -116,7 +116,7 @@ def wrap_data_loader( ) -class BatchMemoryManager(object): +class BatchMemoryManager: """ Context manager to manage memory consumption during training.