From b1bd950a4d3a06b4e105cb77238fb7df01fad4da Mon Sep 17 00:00:00 2001 From: prajjwal1 Date: Mon, 9 Mar 2020 08:43:07 -0700 Subject: [PATCH] Fixed stub for AdamW (#34299) Summary: Fixes [https://github.com/pytorch/pytorch/issues/33757](https://github.com/pytorch/pytorch/issues/33757) Pull Request resolved: https://github.com/pytorch/pytorch/pull/34299 Differential Revision: D20337844 Pulled By: ezyang fbshipit-source-id: 54bf174a09b8db9bf6e0c3c717730dd7c795d76b --- torch/optim/__init__.pyi | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch/optim/__init__.pyi b/torch/optim/__init__.pyi index 85892c28581..e82b5821e5c 100644 --- a/torch/optim/__init__.pyi +++ b/torch/optim/__init__.pyi @@ -3,7 +3,7 @@ from .adadelta import Adadelta from .adagrad import Adagrad from .adam import Adam as Adam from .adamax import Adamax -from .adamw import AdamW +from .adamw import AdamW as AdamW from .asgd import ASGD from .lbfgs import LBFGS from .optimizer import Optimizer