From ad935e65890ba1c00acd4bbc3e33f37bb1067de5 Mon Sep 17 00:00:00 2001 From: Stefano Woerner Date: Tue, 9 May 2023 13:10:28 +0200 Subject: [PATCH] Fix overloading of extract_state_dict --- torchopt/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchopt/utils.py b/torchopt/utils.py index 05867b91..4cd184bb 100644 --- a/torchopt/utils.py +++ b/torchopt/utils.py @@ -99,6 +99,7 @@ def extract_state_dict( by: CopyMode = 'reference', device: Device | None = None, with_buffers: bool = True, + detach_buffers: bool = False, enable_visual: bool = False, visual_prefix: str = '', ) -> ModuleState: # pragma: no cover @@ -111,7 +112,6 @@ def extract_state_dict( *, by: CopyMode = 'reference', device: Device | None = None, - with_buffers: bool = True, enable_visual: bool = False, visual_prefix: str = '', ) -> tuple[OptState, ...]: # pragma: no cover pFad - Phonifier reborn

Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.

Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.


Alternative Proxies:

Alternative Proxy

pFad Proxy

pFad v3 Proxy

pFad v4 Proxy