Skip to content

Commit

Permalink
[Demos] Fix broken mnist source in paddlepaddle demo
Browse files Browse the repository at this point in the history
  • Loading branch information
ClawSeven authored and volcano0dr committed Jun 13, 2024
1 parent 814b573 commit be4de47
Showing 1 changed file with 2 additions and 14 deletions.
16 changes: 2 additions & 14 deletions demos/paddlepaddle/install_python_with_conda.sh
Original file line number Diff line number Diff line change
Expand Up @@ -21,19 +21,7 @@ sed -i "186 i \ elif sysstr == 'occlum':\n return True" $CORE_PY
sed -ie "37,64d" $IMAGE_PY
sed -i "37 i \try:\n import cv2\nexcept ImportError:\n cv2 = None" $IMAGE_PY

# Download the dataset
DATASET=$script_dir/mnist

[ -d $DATASET ] && exit 0

TRAIN_IMAGE=train-images-idx3-ubyte.gz
TRAIN_LABEL=train-labels-idx1-ubyte.gz
TEST_IMAGE=t10k-images-idx3-ubyte.gz
TEST_LABEL=t10k-labels-idx1-ubyte.gz
URL=http://yann.lecun.com/exdb/mnist
# Download the dataset
git clone https://github.com/fgnt/mnist.git

mkdir $DATASET
wget $URL/$TRAIN_IMAGE -P $DATASET
wget $URL/$TRAIN_LABEL -P $DATASET
wget $URL/$TEST_IMAGE -P $DATASET
wget $URL/$TEST_LABEL -P $DATASET

0 comments on commit be4de47

Please sign in to comment.