From bd8f2eb521a99cddbeed45c9935fefd80b5f7ded Mon Sep 17 00:00:00 2001 From: Kaushik Bokka Date: Tue, 16 Feb 2021 13:45:48 +0530 Subject: [PATCH] fix anchor typ --- flash/vision/detection/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/flash/vision/detection/model.py b/flash/vision/detection/model.py index deef62a8f1..e8759751dc 100644 --- a/flash/vision/detection/model.py +++ b/flash/vision/detection/model.py @@ -76,7 +76,7 @@ def __init__( pretrained: bool = True, pretrained_backbone: bool = True, trainable_backbone_layers: int = 3, - achnor_generator: Optional[Type[AnchorGenerator]] = None, + anchor_generator: Optional[Type[AnchorGenerator]] = None, loss=None, metrics: Union[Callable, nn.Module, Mapping, Sequence, None] = None, optimizer: Type[Optimizer] = torch.optim.Adam,