diff --git a/patchgan/train.py b/patchgan/train.py
index 00e7d03..24b304e 100644
--- 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
index 2a7d6fd..00b5ae0 100644
--- 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