This commit is contained in:
parent
b14cc2cce5
commit
b49fa4c2b7
|
@ -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
|
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue
Block a user