diff --git a/.gitignore b/.gitignore
index f8a0d5a..6690668 100644
--- a/.gitignore
+++ b/.gitignore
@@ -2,4 +2,7 @@
 build/
 .vscode/
 *.bin
-*.pyc
\ No newline at end of file
+*.pyc
+*.prototxt
+*.caffemodel
+*.h5
diff --git a/tests/build_models.sh b/tests/build_models.sh
index b3db436..c6b5fe2 100755
--- a/tests/build_models.sh
+++ b/tests/build_models.sh
@@ -1,4 +1,11 @@
 #!/bin/bash
+if [ "$1" == "download" ]; then
+    wget https://github.com/ceccocats/tkDNN/releases/download/testData/tkDNN_testwg.tar.gz --no-check-certificate
+    tar -xf tkDNN_testwg.tar.gz
+    rm tkDNN_testwg.tar.gz
+    exit
+fi
+
 echo "build test Model"
 cd test
 python test_model.py