diff --git a/.gitignore b/.gitignore
index c185d72..ff2718f 100644
--- a/.gitignore
+++ b/.gitignore
@@ -2,7 +2,6 @@
 
 # Byte-compiled / optimized / DLL files
 __pycache__/
-*.h5
 *.ckpt
 *.ipynb
 *.onnx
\ No newline at end of file
diff --git a/data/dataset/train.h5 b/data/dataset/train.h5
new file mode 100644
index 0000000..a894813
Binary files /dev/null and b/data/dataset/train.h5 differ
diff --git a/data/dataset/val.h5 b/data/dataset/val.h5
new file mode 100644
index 0000000..6300e06
Binary files /dev/null and b/data/dataset/val.h5 differ
diff --git a/data/training/train.py b/data/training/train.py
index faf1051..eaa718a 100644
--- a/data/training/train.py
+++ b/data/training/train.py
@@ -1,5 +1,8 @@
 import sys, os
 
+os.environ["NNPACK"] = "0"
+
+
 script_dir = os.path.dirname(os.path.abspath(__file__))
 data_dir = os.path.abspath(os.path.join(script_dir, ".."))
 project_root = os.path.abspath(os.path.join(script_dir, "../.."))