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 f4648b0 commit d200d99Copy full SHA for d200d99
pytorch_optimizer/optimizer/prodigy.py
@@ -111,6 +111,7 @@ def step(self, closure: CLOSURE = None) -> LOSS:
111
if 'd_numerator' not in group:
112
group['d_numerator'] = torch.tensor([0.0], device=device)
113
elif group['d_numerator'].device != device:
114
+ # pragma: no cover
115
group['d_numerator'] = group['d_numerator'].to(device)
116
117
d_numerator = group['d_numerator']
0 commit comments