We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 75dee3a commit 1a0786bCopy full SHA for 1a0786b
pytorch_optimizer/types.py
@@ -6,6 +6,6 @@
6
LOSS = Optional[float]
7
BETAS = Tuple[float, float]
8
DEFAULTS = Dict[str, Any]
9
-PARAMETERS = Union[Iterable[Dict[str, Any]], Iterable[torch.Tensor]]
+PARAMETERS = Optional[Union[Iterable[Dict[str, Any]], Iterable[torch.Tensor]]]
10
STATE = Dict[str, Any]
11
BUFFER = List[List[Optional[torch.Tensor]]]
0 commit comments