qoherent/modrec-workflow
14
0
Fork 1
Code Issues 1 Pull Requests Workflows Packages Projects Releases 1 Wiki 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 # Byte-compiled / optimized / DLL files
__pycache__/ __pycache__/
*.h5
*.ckpt *.ckpt
*.ipynb *.ipynb
*.onnx *.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 import sys, os
os.environ["NNPACK"] = "0"
script_dir = os.path.dirname(os.path.abspath(__file__)) script_dir = os.path.dirname(os.path.abspath(__file__))
data_dir = os.path.abspath(os.path.join(script_dir, "..")) data_dir = os.path.abspath(os.path.join(script_dir, ".."))
project_root = os.path.abspath(os.path.join(script_dir, "../..")) project_root = os.path.abspath(os.path.join(script_dir, "../.."))