qoherent/modrec-workflow
14
0
Fork 1
Code Issues 1 Pull Requests Workflows Packages Projects Releases 1 Wiki Activity

changed the training to only do 5 epochs, added in a convert to ORT at the end of worklflow
Some checks failed
RIA Hub Workflow Demo / ria-demo (push) Has been cancelled
Details

Browse Source
This commit is contained in:
Liyu Xiao 2025-05-26 09:43:42 -04:00
parent 7e255a704d
commit 015f6d4db5
2 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
data/training/train.py
View File

@ -27,6 +27,7 @@ from modulation_dataset import ModulationH5Dataset
import mobilenetv3
class CleanProgressCallback(Callback):
"""Clean progress callback that only shows epoch summaries"""