(self, data, device)
| 109 | self.buffer = io.BytesIO() |
| 110 | |
| 111 | def to_device(self, data, device): |
| 112 | if isinstance(data, dict): |
| 113 | return {key: self.to_device(value, device) for key, value in data.items()} |
| 114 | elif isinstance(data, list): |
| 115 | return [self.to_device(item, device) for item in data] |
| 116 | elif isinstance(data, torch.Tensor): |
| 117 | return data.to(device) |
| 118 | else: |
| 119 | return data |
| 120 | |
| 121 | def prepare_tensor(self, data) -> bytes: |
| 122 | self.buffer.seek(0) |
no test coverage detected