From cb05de36f5ffd8584d78c6776dbe90e21abff25a Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Sun, 4 Apr 2021 12:44:40 +0800 Subject: Remove triplet sampler --- utils/configuration.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'utils/configuration.py') diff --git a/utils/configuration.py b/utils/configuration.py index fff3876..1ace241 100644 --- a/utils/configuration.py +++ b/utils/configuration.py @@ -25,7 +25,7 @@ class DatasetConfiguration(TypedDict): class DataloaderConfiguration(TypedDict): - batch_size: tuple[int, int] + batch_size: int num_workers: int pin_memory: bool -- cgit v1.2.3