From eb04c4f268d276fdc6b96aa8ae86407b3eba9dc4 Mon Sep 17 00:00:00 2001
From: ItsThanhTung <84335584+ItsThanhTung@users.noreply.github.com>
Date: Mon, 6 Nov 2023 13:25:04 +0700
Subject: [PATCH] Minor bug in evaluation_quantitative.py

---
 evaluation_quantitative.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/evaluation_quantitative.py b/evaluation_quantitative.py
index caebedd..1bf444c 100644
--- a/evaluation_quantitative.py
+++ b/evaluation_quantitative.py
@@ -49,9 +49,9 @@ def main(cfg: DictConfig):
         length = len(generator.dataset)
 
     if "srn" in cfg.model_path:
-        non_one_length = True
-    else:
         non_one_length = False
+    else:
+        non_one_length = True
 
     chunks = num_to_chunks(length, cfg.eval.n_devices)
     start_idx = sum(chunks[:cfg.eval.device_idx])
@@ -191,4 +191,4 @@ def num_to_chunks(num, groups):
     return arr
 
 if __name__ == "__main__":
-    main()
\ No newline at end of file
+    main()