From 5295f82637428595658274d916f422ec85ae4d04 Mon Sep 17 00:00:00 2001 From: Jason Senthil Date: Tue, 6 Feb 2024 10:42:39 -0800 Subject: [PATCH] fix GradScaler pre-commit (#699) Summary: Pull Request resolved: https://github.com/pytorch/tnt/pull/699 Fixing failed pre-commit message here: {F1452790112} Reviewed By: gunchu Differential Revision: D53456990 fbshipit-source-id: 7b3b87f51aa16ac4eeb76ada8fbf755a55ac2967 --- 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