def collect_activations(
        model_path: str,
        layers: List[str],
        dataset: str,
        data_format: str = None,
        split: str = FULL,
        batch_size: int = 128,
        output_directory: str = 'results',
        gpus: List[str] = None,
        gpu_memory_limit: int =None,
        allow_parallel_threads: bool = True,
        backend: Union[Backend, str] = None,
        debug: bool = False,
        **kwargs