From f12731e36550be2d0b97b052eb33f2272f339454 Mon Sep 17 00:00:00 2001 From: Oliver Borchert Date: Tue, 21 Dec 2021 02:24:34 +0100 Subject: [PATCH] Fix data loader init --- lightkit/data/loader.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/lightkit/data/loader.py b/lightkit/data/loader.py index 8ae5146..1055025 100644 --- a/lightkit/data/loader.py +++ b/lightkit/data/loader.py @@ -42,6 +42,9 @@ def __init__(self, dataset: Dataset[T_co], **kwargs: Any): batch_size=kwargs.get("batch_size", 1), drop_last=kwargs.get("drop_last", False), ) + kwargs.pop("batch_size", None) + kwargs.pop("shuffle", None) + kwargs.pop("drop_last", None) kwargs.setdefault("collate_fn", collate_tuple) super().__init__(dataset, **kwargs) # type: ignore