mirror of
https://github.com/wassname/dreamerv3-torch.git
synced 2026-06-27 16:15:31 +08:00
avoid mutable default argument
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user