From 97c8ab79642208925e31f3e844570562bd743ef9 Mon Sep 17 00:00:00 2001 From: dkloving Date: Tue, 18 May 2021 21:08:41 -0500 Subject: [PATCH] Made targets argument in YOLOTransform.forward() truly optional. (#108) `targets` was missing a default value of `None` making using YOLOTransform on its own require explicitly passing `None` during inference. --- yolort/models/transform.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/yolort/models/transform.py b/yolort/models/transform.py index 2d3b9878..dcd090dc 100644 --- a/yolort/models/transform.py +++ b/yolort/models/transform.py @@ -55,7 +55,7 @@ def __init__(self, min_size, max_size) -> None: def forward( self, images: List[Tensor], - targets: Optional[List[Dict[str, Tensor]]], + targets: Optional[List[Dict[str, Tensor]]] = None, ) -> Tuple[NestedTensor, Optional[Tensor]]: device = images[0].device images = [img for img in images]