(dataset_kwargs_list: List[Dict[str, Any]], dataset_weights: List[int])
| 689 | |
| 690 | # === RLDS Dataset Initialization Utilities === |
| 691 | def pprint_data_mixture(dataset_kwargs_list: List[Dict[str, Any]], dataset_weights: List[int]) -> None: |
| 692 | print("\n######################################################################################") |
| 693 | print(f"# Loading the following {len(dataset_kwargs_list)} datasets (incl. sampling weight):{'': >24} #") |
| 694 | for dataset_kwargs, weight in zip(dataset_kwargs_list, dataset_weights): |
| 695 | pad = 80 - len(dataset_kwargs["name"]) |
| 696 | print(f"# {dataset_kwargs['name']}: {weight:=>{pad}f} #") |
| 697 | print("######################################################################################\n") |
| 698 | |
| 699 | |
| 700 | def get_dataset_statistics( |
no outgoing calls
no test coverage detected