-
Notifications
You must be signed in to change notification settings - Fork 0
/
travis.sh
executable file
·45 lines (37 loc) · 932 Bytes
/
travis.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
#!/usr/bin/env bash
set -e
./generatepb.sh
# build
mkdir build && cd build
cmake .. -DUSE_JAVA=ON -DBLAS=blas -DCMAKE_BUILD_TYPE=Release
make
# before test
URL_model="https://github.com/luoyetx/misc/blob/master/mini-caffe/model.zip?raw=true"
if [ ! -f model.zip ]; then
echo "Downloading model.zip from $URL_model"
wget -O model.zip $URL_model
fi
unzip -o model.zip
# test
./run_net
./run_net_c
./benchmark ./model/resnet.prototxt 1 -1
cd ..
# java test
cd java
./gradlew clean build --info
cd ..
# python test
cd python
python2 --version
python2 tests/test.py
python2 setup.py build
python2 setup.py clean
python3 --version
python3 tests/test.py
python3 setup.py build
python3 setup.py clean
cd ..
# test fuse_bn
#python2 tools/fuse_bn.py --net ./build/model/resnet.prototxt --weight ./build/model/resnet.caffemodel
#python3 tools/fuse_bn.py --net ./build/model/resnet.prototxt --weight ./build/model/resnet.caffemodel