(data_directory, index)
| 55 | |
| 56 | |
| 57 | def get_random_partition(data_directory, index): |
| 58 | partitions = [os.path.join(data_directory, x) |
| 59 | for x in os.listdir(data_directory)] |
| 60 | partitions = sorted(partitions) |
| 61 | i = index % len(partitions) |
| 62 | return partitions[i] |
| 63 | |
| 64 | |
| 65 | def map_to_torch(encoding): |