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

updated conf

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: 4
#output of the onnx model
onnx_model_filename: inference_recognition_model
app:
build_dir: dist

3
helpers/app_settings.py
View File

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