Function load_calibration_dataset

ppq/samples/quantize_caffe_model.py:17–18

()

MODEL_PATH = 'Models/model.caffemodel' # For successfully loading caffe model, .caffemodel file is required.

def load_calibration_dataset() -> Iterable:
    return [torch.rand(size=INPUT_SHAPE) for _ in range(32)]

def collate_fn(batch: torch.Tensor) -> torch.Tensor:
    return batch.to(DEVICE)