diff --git a/orcasong/tools/shuffle2.py b/orcasong/tools/shuffle2.py index 28f8a888ec7344a4a76795577e69df70df8ca338..90859dc4412998e5cc21a85e2a7d69882fc592bb 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