snsd0805/Distributed-Training-Example
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: delete matplotlib dependence

Browse Source
This commit is contained in:
TING-JUN WANG 2024-05-16 23:09:35 +08:00
parent 233bec6d1c
commit 874c160eae
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
train/train.py
View File

@ -5,7 +5,6 @@ from torch import nn
from dataset import Cifar10Dataset from dataset import Cifar10Dataset
from model import Network from model import Network
from torch.utils.data import DataLoader from torch.utils.data import DataLoader
import matplotlib.pyplot as plt
import torch.multiprocessing as mp import torch.multiprocessing as mp
from torch.utils.data.distributed import DistributedSampler from torch.utils.data.distributed import DistributedSampler