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

updated conf
Some checks failed
RIA Hub Workflow Demo / ria-demo (push) Failing after 24s
Details

Browse Source
This commit is contained in:
Liyu Xiao 2025-06-13 14:00:27 -04:00
parent b14cc2cce5
commit b49fa4c2b7
2 changed files with 0 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
conf/app.yaml
View File

@ -39,8 +39,6 @@ inference:
#num classes to classify on #num classes to classify on
num_classes: 4 num_classes: 4
#output of the onnx model
onnx_model_filename: inference_recognition_model
app: app:
build_dir: dist build_dir: dist

3
helpers/app_settings.py
View File

@ -26,15 +26,12 @@ class TrainingConfig:
batch_size: int batch_size: int
epochs: int epochs: int
learning_rate: float learning_rate: float
checkpoint_dir: str
checkpoint_filename: str
use_gpu: bool use_gpu: bool
@dataclass @dataclass
class InferenceConfig: class InferenceConfig:
num_classes: int num_classes: int
onnx_model_filename: str
@dataclass @dataclass