From d11089079e4e631f8c0e78012ecd74d11bddd266 Mon Sep 17 00:00:00 2001 From: plusk-dev Date: Thu, 17 Oct 2024 00:36:58 +0530 Subject: [PATCH] removed ipdb unused import --- dfme/network/gan.py | 1 - dfme/train.py | 2 +- surrogate_benchmark/dataloader.py | 1 - surrogate_benchmark/train.py | 2 -- 4 files changed, 1 insertion(+), 5 deletions(-) diff --git a/dfme/network/gan.py b/dfme/network/gan.py index df809ad..8383ee6 100644 --- a/dfme/network/gan.py +++ b/dfme/network/gan.py @@ -1,7 +1,6 @@ import torch import torch.nn as nn import torch.nn.functional as F -import ipdb class Flatten(nn.Module): def __init__(self): diff --git a/dfme/train.py b/dfme/train.py index a7ca4ee..4d499bf 100644 --- a/dfme/train.py +++ b/dfme/train.py @@ -1,5 +1,5 @@ from __future__ import print_function -import argparse, ipdb, json +import argparse, json import torch import torch.nn.functional as F import torch.nn as nn diff --git a/surrogate_benchmark/dataloader.py b/surrogate_benchmark/dataloader.py index 02f1149..df1ec44 100644 --- a/surrogate_benchmark/dataloader.py +++ b/surrogate_benchmark/dataloader.py @@ -1,7 +1,6 @@ from torchvision import datasets, transforms from torch.utils.data import DataLoader, Dataset import PIL -import ipdb import torch class RandDataset(Dataset): diff --git a/surrogate_benchmark/train.py b/surrogate_benchmark/train.py index 9df939a..dbc6498 100644 --- a/surrogate_benchmark/train.py +++ b/surrogate_benchmark/train.py @@ -1,5 +1,4 @@ import torch -import ipdb import torch.nn as nn import torch.optim as optim import sys, os, json @@ -38,7 +37,6 @@ def epoch(args, loader, model, teacher = None, lr_schedule = None, epoch_i = Non func = tqdm if stop == False else lambda x:x criterion_kl = nn.KLDivLoss(reduction = "batchmean") alpha, T = 1.0, args.temp - # ipdb.set_trace() for batch in func(loader): X,y = batch[0].to(args.device), batch[1].to(args.device) if args.surrogate == "mnist":