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