update: AdaBound

kozistr · kozistr · commit 9b4f5facc813 · 2021-09-22T15:41:27.000+09:00
diff --git a/pytorch_optimizer/__init__.py b/pytorch_optimizer/__init__.py
@@ -1,4 +1,4 @@
-from pytorch_optimizer.adabound import AdaBoundW
+from pytorch_optimizer.adabound import AdaBound, AdaBoundW
 from pytorch_optimizer.adahessian import AdaHessian
 from pytorch_optimizer.adamp import AdamP
 from pytorch_optimizer.agc import agc

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-from pytorch_optimizer.adabound import AdaBoundW`
	`1`	`+from pytorch_optimizer.adabound import AdaBound, AdaBoundW`
`2`	`2`	`from pytorch_optimizer.adahessian import AdaHessian`
`3`	`3`	`from pytorch_optimizer.adamp import AdamP`
`4`	`4`	`from pytorch_optimizer.agc import agc`