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

changed the training to only do 5 epochs, added in a convert to ORT at the end of worklflow

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 import mobilenetv3
class CleanProgressCallback(Callback): class CleanProgressCallback(Callback):
"""Clean progress callback that only shows epoch summaries""" """Clean progress callback that only shows epoch summaries"""