更新 'gpu_mnist_example/parallel_train.py' · 885820fa43 - OpenI_Cloudbrain_Example - 开源协同云脑生态支撑系统

更新 'gpu_mnist_example/parallel_train.py'

liuzx

liuzx 1 year ago

parent

3d9d0ba397

commit

885820fa43

1 changed files with 1 additions and 0 deletions

Unified View

							
								+ 1
								
								- 0
							
gpu_mnist_example/parallel_train.pyView File
								
																@@ -28,6 +28,7 @@ from torch.utils.data import DataLoader
															
																from torchvision.transforms import ToTensor
																
																from torchvision.transforms import ToTensor
															
																import argparse
																
																import argparse
															
																import os
																
																import os
															
																os.system("pip install c2net")
															
																#导入c2net包
																
																#导入c2net包
															
																from c2net.context import prepare
																
																from c2net.context import prepare