From edd28359ed91396384ee12e722fcc3c34a06ed6d Mon Sep 17 00:00:00 2001 From: Kajetan Schweighofer Date: Thu, 4 May 2023 14:21:37 +0200 Subject: [PATCH] Update __init__.py To conviniently import the ABAnnealing scheduler through from torch_sgld import ABAnnealingLR --- torch_sgld/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/torch_sgld/__init__.py b/torch_sgld/__init__.py index 4b8c279..fdb011b 100644 --- a/torch_sgld/__init__.py +++ b/torch_sgld/__init__.py @@ -1,8 +1,8 @@ from .sgld import SGLD -from .lr_scheduler import CosineLR +from .lr_scheduler import CosineLR, ABAnnealingLR __all__ = [ 'SGLD', 'ABAnnealingLR', 'CosineLR', -] \ No newline at end of file +]