diff --git a/configs/model/depth.yaml b/configs/model/depth.yaml index bfeb4b8..b9d9d01 100644 --- a/configs/model/depth.yaml +++ b/configs/model/depth.yaml @@ -13,7 +13,7 @@ scheduler: factor: 0.1 threshold: 0.0001 patience: 5 - threshold_mode: 'abs' + threshold_mode: "abs" net: _target_: src.models.components.depth_net_efficient_ffn.DepthNet diff --git a/configs/quantization.yaml b/configs/quantization.yaml index 705b8dd..63acd70 100644 --- a/configs/quantization.yaml +++ b/configs/quantization.yaml @@ -20,10 +20,10 @@ save_path: name.pty quantizer: config: asymmetric: true - backend: 'qnnpack' + backend: "qnnpack" disable_requantization_for_cat: true per_tensor: false - work_dir: 'quant_output' + work_dir: "quant_output" task_name: "quantization"