From 0ba839dec4486485f90fc18239559fb749236afc Mon Sep 17 00:00:00 2001 From: Jason Senthil Date: Mon, 5 Feb 2024 17:42:40 -0800 Subject: [PATCH] fix GradScaler pre-commit Summary: Fixing failed pre-commit message here: {F1452790112} Differential Revision: D53456990 --- torchtnt/framework/auto_unit.py | 1 - 1 file changed, 1 deletion(-) diff --git a/torchtnt/framework/auto_unit.py b/torchtnt/framework/auto_unit.py index ce4412adb3..db720c1560 100644 --- a/torchtnt/framework/auto_unit.py +++ b/torchtnt/framework/auto_unit.py @@ -13,7 +13,6 @@ import torch from pyre_extensions import none_throws -from torch.cuda.amp import GradScaler from torch.distributed.fsdp import FullyShardedDataParallel as FSDP from torch.nn.parallel import DistributedDataParallel as DDP from torch.optim.swa_utils import SWALR