(self, outputs, output_device)
| 166 | [kwargs] * len(devices) |
| 167 | |
| 168 | def gather(self, outputs, output_device): |
| 169 | out = {} |
| 170 | |
| 171 | for k in outputs[0]: |
| 172 | out[k] = torch.stack([output[k].to(output_device) for output in outputs]) |
| 173 | |
| 174 | return out |
| 175 | |
| 176 | @torch.no_grad() |
| 177 | def prepare_data(self, datum, devices:list=None, allocation:list=None): |
nothing calls this directly
no outgoing calls
no test coverage detected