From 874c160eae6223a64abf647a9bfdbd912bd810c6 Mon Sep 17 00:00:00 2001 From: TING-JUN WANG Date: Thu, 16 May 2024 23:09:35 +0800 Subject: [PATCH] fix: delete matplotlib dependence --- train/train.py | 1 - 1 file changed, 1 deletion(-) 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