From 9fc188c254c57f6aec04557fbbce928e0d1be97a Mon Sep 17 00:00:00 2001
From: Stefan Reck <stefan.reck@fau.de>
Date: Tue, 6 Jul 2021 12:42:53 +0200
Subject: [PATCH] minor
---
orcasong/tools/shuffle2.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/orcasong/tools/shuffle2.py b/orcasong/tools/shuffle2.py
index 28f8a88..90859dc 100644
--- a/orcasong/tools/shuffle2.py
+++ b/orcasong/tools/shuffle2.py
@@ -146,8 +146,9 @@ def get_n_iterations(
n_iterations = int(
np.ceil(np.log(dset_info["n_chunks"]) / np.log(dset_info["chunks_per_batch"]))
)
+ print(f"Largest dataset: {dset_info['name']}")
print(f"Total chunks: {dset_info['n_chunks']}")
- print(f"Chunks per batch: {dset_info['chunks_per_batch']}")
+ print(f"Max. chunks per batch: {dset_info['chunks_per_batch']}")
print(f"--> min iterations for full shuffle: {n_iterations}")
return n_iterations
--
GitLab