hongshaorou/dreamerv3-torch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

avoid mutable default argument

Browse Source
This commit is contained in:
NM512 2024-03-11 06:21:35 +09:00
parent 2c7a81a0e2
commit 2cfcaefea2
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tools.py
View File

@ -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