update: RuntimeError message

kozistr · kozistr · commit fea55f27ea5d · 2021-09-22T16:36:10.000+09:00
diff --git a/pytorch_optimizer/adabelief.py b/pytorch_optimizer/adabelief.py
@@ -125,7 +125,7 @@ def step(self, closure: CLOSURE = None) -> LOSS:
                 grad = p.grad.data
                 if grad.is_sparse:
                     raise RuntimeError(
-                        'AdaBelief does not support sparse gradients, please consider SparseAdam instead'
+                        'AdaBelief does not support sparse gradients'
                     )
 
                 amsgrad = group['amsgrad']

Original file line number	Diff line number	Diff line change
`@@ -125,7 +125,7 @@ def step(self, closure: CLOSURE = None) -> LOSS:`
`125`	`125`	`grad = p.grad.data`
`126`	`126`	`if grad.is_sparse:`
`127`	`127`	`raise RuntimeError(`
`128`		`- 'AdaBelief does not support sparse gradients, please consider SparseAdam instead'`
	`128`	`+ 'AdaBelief does not support sparse gradients'`
`129`	`129`	`)`
`130`	`130`
`131`	`131`	`amsgrad = group['amsgrad']`