michael/modrec-workflow
1
0
Fork 0
forked from qoherent/modrec-workflow
Code Pull Requests Workflows Activity

got

Browse Source
This commit is contained in:
Liyu Xiao 2025-05-22 16:12:02 -04:00
parent ecd039b336
commit f3fd1f7c59
4 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@ -2,7 +2,6 @@
# Byte-compiled / optimized / DLL files
__pycache__/
*.h5
*.ckpt
*.ipynb
*.onnx

BIN
data/dataset/train.h5 Normal file
View File

Binary file not shown.

BIN
data/dataset/val.h5 Normal file
View File

Binary file not shown.

3
data/training/train.py
View File

@ -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, "../.."))