(start=None, end=None, cfg=None)
| 113 | |
| 114 | |
| 115 | def GenerateRunSet(start=None, end=None, cfg=None): |
| 116 | global dataset, _maxRuns |
| 117 | if start is None: |
| 118 | start = 0 |
| 119 | if end is None: |
| 120 | end = _maxRuns |
| 121 | if cfg is None: |
| 122 | cfg = {"shot": 1, "ways": 5, "queries": 15} |
| 123 | |
| 124 | setRandomStates(cfg) |
| 125 | print("generating task from {} to {}".format(start, end)) |
| 126 | |
| 127 | dataset = [] |
| 128 | for iRun in range(end - start): |
| 129 | dataset.append(GenerateRun(start + iRun, cfg)) |
| 130 | |
| 131 | return torch.stack(dataset) |
nothing calls this directly
no test coverage detected