()
| 18 | PLATFORM = TargetPlatform.PPL_CUDA_INT8 # identify a target platform for your network. |
| 19 | |
| 20 | def load_calibration_dataset() -> Iterable: |
| 21 | return [torch.rand(size=INPUT_SHAPE) for _ in range(32)] |
| 22 | |
| 23 | def collate_fn(batch: torch.Tensor) -> torch.Tensor: |
| 24 | return batch.to(DEVICE) |