From 70e6705027e72ef51639d0ddd979fd542d75e672 Mon Sep 17 00:00:00 2001 From: mattiasakesson Date: Mon, 2 Sep 2024 13:51:58 +0200 Subject: [PATCH] remove un. package --- examples/mnist-pytorch-DPSGD/client/train.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/examples/mnist-pytorch-DPSGD/client/train.py b/examples/mnist-pytorch-DPSGD/client/train.py index b96de7afb..4a54abc51 100644 --- a/examples/mnist-pytorch-DPSGD/client/train.py +++ b/examples/mnist-pytorch-DPSGD/client/train.py @@ -10,7 +10,6 @@ from opacus import PrivacyEngine from torch.utils.data import Dataset, DataLoader -from tqdm.notebook import tqdm import numpy as np from opacus.utils.batch_memory_manager import BatchMemoryManager @@ -107,7 +106,7 @@ def train(in_model_path, out_model_path, data_path=None, batch_size=32, epochs=1 - for epoch in tqdm(range(EPOCHS), desc="Epoch", unit="epoch"): + for epoch in range(EPOCHS): train_dp(model, train_loader, optimizer, epoch + 1, device, privacy_engine) d_epsilon = privacy_engine.get_epsilon(DELTA)