(batch: dict)
| 29 | return dataset, sample # last sample |
| 30 | |
| 31 | def collate_fn(batch: dict) -> torch.Tensor: |
| 32 | return {k: v.to(DEVICE) for k, v in batch.items()} |
| 33 | |
| 34 | # ------------------------------------------------------------ |
| 35 | # 在这里，我们仍然创建一个 QuantizationSetting 对象用来管理量化过程 |