From 2cfcaefea21de58e98e3a13771c4f6105efede13 Mon Sep 17 00:00:00 2001 From: NM512 Date: Mon, 11 Mar 2024 06:21:35 +0900 Subject: [PATCH] avoid mutable default argument --- tools.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tools.py b/tools.py index f80384e..87f2633 100644 --- a/tools.py +++ b/tools.py @@ -963,10 +963,12 @@ def enable_deterministic_run(): def recursively_collect_optim_state_dict( - obj, path="", optimizers_state_dicts=None, visited=set() + obj, path="", optimizers_state_dicts=None, visited=None ): if optimizers_state_dicts is None: optimizers_state_dicts = {} + if visited is None: + visited = set() # avoid cyclic reference if id(obj) in visited: return optimizers_state_dicts