Andreas Kirsch

icon indicating a website link blackhc.net icon indicating an email [email protected]

icon company @OATML icon location Oxford, UK icon location Past: AIMS DPhil in Oxford at @OATML; Fellow at @nwspk; RE at DeepMind; SWE at Google.

Results 22 comments of Andreas Kirsch

Support downloading specific splits in `load_dataset`

No worries! I've patched the ImageNet dataset in: Together with: ```python dataset = load_dataset( "ILSVRC/imagenet-1k", split="validation", data_files={"val": "data/val_images.tar.gz"}, revision="refs/pr/20", trust_remote_code=True, download_config=DownloadConfig(resume_download=True), verification_mode=VerificationMode.NO_CHECKS, ) ``` It only downloads the validation set...

Feature Request: Allow sampling of log probs and logits for Likelihood.CLASSIFICATION

Aww, yeah, that would be great! It would cover all my use cases and provide a nice extensible interface.