From b49fa4c2b76fdfd783db8834202c4f4ab0e7a0cb Mon Sep 17 00:00:00 2001 From: Liyu Xiao Date: Fri, 13 Jun 2025 14:00:27 -0400 Subject: [PATCH] updated conf --- conf/app.yaml | 2 -- helpers/app_settings.py | 3 --- 2 files changed, 5 deletions(-) diff --git a/conf/app.yaml b/conf/app.yaml index ed739cf..a637988 100644 --- a/conf/app.yaml +++ b/conf/app.yaml @@ -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 \ No newline at end of file diff --git a/helpers/app_settings.py b/helpers/app_settings.py index a53a9d2..cd91d8d 100644 --- a/helpers/app_settings.py +++ b/helpers/app_settings.py @@ -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