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()