diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/examples/mnist/setup.sh | 12 |
1 files changed, 5 insertions, 7 deletions
diff --git a/tests/examples/mnist/setup.sh b/tests/examples/mnist/setup.sh index 8dead6a..1d0221f 100644 --- a/tests/examples/mnist/setup.sh +++ b/tests/examples/mnist/setup.sh @@ -6,14 +6,12 @@ set -e # Setup code goes here test -e $MRT_MARIAN/mnist_example -(( $(ls -1 *-ubyte 2>/dev/null | wc -l) == 4 )) && exit 0 +test -e $MRT_DATA/exdb_mnist/train-images-idx3-ubyte +test -e $MRT_DATA/exdb_mnist/train-labels-idx1-ubyte +test -e $MRT_DATA/exdb_mnist/t10k-images-idx3-ubyte +test -e $MRT_DATA/exdb_mnist/t10k-labels-idx1-ubyte -wget -nc http://yann.lecun.com/exdb/mnist/train-images-idx3-ubyte.gz -wget -nc http://yann.lecun.com/exdb/mnist/train-labels-idx1-ubyte.gz -wget -nc http://yann.lecun.com/exdb/mnist/t10k-images-idx3-ubyte.gz -wget -nc http://yann.lecun.com/exdb/mnist/t10k-labels-idx1-ubyte.gz - -gzip -d *-ubyte.gz +test -e *-ubyte || cp $MRT_DATA/exdb_mnist/*-ubyte . # Exit with success code exit 0 |