fix: delete matplotlib dependence

2024-05-16 23:09:35 +08:00 · 2024-05-16 23:09:35 +08:00 · 874c160eae
commit 874c160eae
parent 233bec6d1c
1 changed files with 0 additions and 1 deletions
--- 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