diff --git a/train/train.py b/train/train.py
index eb8f4f4..d8b409b 100644
--- a/train/train.py
+++ b/train/train.py
@@ -5,7 +5,6 @@ from torch import nn
 from dataset import Cifar10Dataset
 from model import Network
 from torch.utils.data import DataLoader
-import matplotlib.pyplot as plt
 
 import torch.multiprocessing as mp
 from torch.utils.data.distributed import DistributedSampler