fixing flake issues

ramanakumars · Nov 21, 2023 · 23fc791 · 23fc791
1 parent 245f0b3
commit 23fc791
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 7 deletions.
diff --git a/patchgan/train.py b/patchgan/train.py
@@ -1,7 +1,5 @@
 import torch
 from torchinfo import summary
-from patchgan.unet import UNet
-from patchgan.disc import Discriminator
 from patchgan.io import COCOStuffDataset
 from patchgan.trainer import PatchGAN
 from torch.utils.data import DataLoader, random_split

diff --git a/patchgan/trainer.py b/patchgan/trainer.py
@@ -1,13 +1,9 @@
 import torch
 import os
-import tqdm
 import glob
-import numpy as np
-from torch import optim
-from torch.optim.lr_scheduler import ExponentialLR, ReduceLROnPlateau
+from torch.optim.lr_scheduler import ExponentialLR
 from .losses import fc_tversky, bce_loss, MAE_loss
 from torch.nn.functional import binary_cross_entropy
-from collections import defaultdict
 from .unet import UNet
 from .disc import Discriminator
 import lightning as L