From c7eee3cdc25454fbb2291c94ef79aba37ace3f05 Mon Sep 17 00:00:00 2001 From: Akemi Izuko Date: Fri, 29 Nov 2024 18:30:26 -0700 Subject: [PATCH] Torchlira: remove secure mode --- lira-pytorch/train.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/lira-pytorch/train.py b/lira-pytorch/train.py index 5c6a538..c28eee1 100644 --- a/lira-pytorch/train.py +++ b/lira-pytorch/train.py @@ -115,7 +115,7 @@ def run(): optim = torch.optim.SGD(m.parameters(), lr=args.lr, momentum=0.9, weight_decay=5e-4) sched = torch.optim.lr_scheduler.CosineAnnealingLR(optim, T_max=args.epochs) - privacy_engine = PrivacyEngine(accountant='rdp', secure_mod=True) + privacy_engine = PrivacyEngine() m, optim, train_dl = privacy_engine.make_private_with_epsilon( module=m, optimizer=optim, @@ -135,7 +135,6 @@ def run(): max_physical_batch_size=1000, optimizer=optim ) as memory_safe_data_loader: - for i in tqdm(range(args.epochs)): m.train() loss_total = 0