minor tidy-up

2026-08-11 11:13:12 +08:00 · 2023-02-08 20:57:13 +00:00
parent 283df8ec84
commit 9faae250ce
2 changed files with 2 additions and 2 deletions
@@ -172,6 +172,7 @@ def get_dataset_name_from_data_config(data_config):


 def get_dataset_fractions(conf, dataset_sizes):
+    """Calculate fraction of each dataset to use per epoch when subsampling"""
    fractions = []
    for i, data_config in enumerate(conf):
        dataset_name = get_dataset_name_from_data_config(data_config)