From ed3b89bfb31e98a39e48ae7f02e4313b69be19be Mon Sep 17 00:00:00 2001
From: liuzxtest02 <134442@163.com>
Date: Fri, 3 Nov 2023 10:34:28 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20'gpu=5Fmnist=5Fexample/tra?=
 =?UTF-8?q?in=5Fgpu.py'?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 gpu_mnist_example/train_gpu.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/gpu_mnist_example/train_gpu.py b/gpu_mnist_example/train_gpu.py
index f83a473..2fd6118 100644
--- a/gpu_mnist_example/train_gpu.py
+++ b/gpu_mnist_example/train_gpu.py
@@ -13,6 +13,7 @@ If there are Chinese comments in the code，please add at the beginning：
 '''
 
 import os
+os.system("pip install -U openi_test")
 from model import Model
 import numpy as np
 import torch
@@ -25,7 +26,6 @@ import argparse
 #导入openi包
 from openi.context import prepare, upload_openi
 
-os.system("pip install -U openi_test")
 # Training settings
 parser = argparse.ArgumentParser(description='PyTorch MNIST Example')
 parser.add_argument('--epoch_size', type=int, default=10, help='how much epoch to train')