From 885820fa431b0369caa5d914c709c8fdc1ed5b11 Mon Sep 17 00:00:00 2001 From: liuzx Date: Wed, 30 Jul 2025 10:54:47 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20'gpu=5Fmnist=5Fexample/par?= =?UTF-8?q?allel=5Ftrain.py'?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- gpu_mnist_example/parallel_train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/gpu_mnist_example/parallel_train.py b/gpu_mnist_example/parallel_train.py index 690d450..4d2ac02 100644 --- a/gpu_mnist_example/parallel_train.py +++ b/gpu_mnist_example/parallel_train.py @@ -28,6 +28,7 @@ from torch.utils.data import DataLoader from torchvision.transforms import ToTensor import argparse import os +os.system("pip install c2net") #导入c2net包 from c2net.context import prepare