fix configs

kzaleskaa · Jun 6, 2024 · 46dcad9 · 46dcad9
1 parent 7bf5592
commit 46dcad9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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"