(config: _config.TrainConfig)
| 123 | |
| 124 | |
| 125 | def build_datasets(config: _config.TrainConfig): |
| 126 | # Use the unified data loader with PyTorch framework |
| 127 | data_loader = _data.create_data_loader(config, framework="pytorch", shuffle=True) |
| 128 | return data_loader, data_loader.data_config() |
| 129 | |
| 130 | |
| 131 | def get_model_state_dict(model): |
no test coverage detected