data:
    batch_size: 32
    dataset_params:
        iq_key: iq_data
        label_key: impairment
        metadata_key: metadata/metadata
        snr_key: Z
    drop_last: true
    kind: iq_h5_auto
    num_workers: 4
    persistent_workers: false
    pin_memory: true
    test_path: /opt/qmb/riahub/dataset/qoherent/pluto-interference-recognition-demo/main/test.h5
    test_split: 0
    train_path: /opt/qmb/riahub/dataset/qoherent/pluto-interference-recognition-demo/main/train.h5
    validation_split: 0
evaluation:
    capture_predictions: true
    enabled: true
    params:
        save_confusion: true
    split: test
export:
    dynamic_batch: true
    dynamic_width: false
    enabled: true
    file_name: model.onnx
    opset_version: 17
    strict: false
    use_dynamo: true
    use_onnxsim: false
model:
    name: iq_mobilenetv3
    params:
        model_size: mobilenetv3_small_050
optimization:
    loss:
        name: cross_entropy
        params: {}
    optimizer:
        name: adam
        params:
            amsgrad: false
            eps: 1e-08
            lr: 0.001
            weight_decay: 0
runtime:
    amp_enabled: false
    autocast_dtype: float32
    checkpoint_every_n_epochs: 1
    component_modules: []
    device: auto
    epochs: 1
    progress_bar: false
    seed: 42
task:
    name: classification
    params:
        save_artifacts: true
        selection_metric: accuracy
        selection_mode: max