(data, data_dist="iid", device_id=None)
| 184 | |
| 185 | |
| 186 | def get_data(data, data_dist="iid", device_id=None): |
| 187 | if data == "bank": |
| 188 | train_x, train_y, val_x, val_y, num_classes = bank.load(device_id) |
| 189 | else: |
| 190 | raise NotImplementedError |
| 191 | return train_x, train_y, val_x, val_y, num_classes |
| 192 | |
| 193 | |
| 194 | def get_model(model, num_channels=None, num_classes=None, data_size=None): |