Function collate_fn

ppq/samples/Tutorial/bestPractice.py:17–18

(batch: torch.Tensor)

CALIBRATION = [torch.rand(size=INPUT_SHAPE) for _ in range(32)]
QS          = QuantizationSettingFactory.default_setting()
def collate_fn(batch: torch.Tensor) -> torch.Tensor:
    return batch.to(DEVICE)

model = torchvision.models.mobilenet.mobilenet_v2(pretrained=True)
model = model.to(DEVICE)