fix anchor typ

Lightning-Universe · Feb 16, 2021 · bd8f2eb · bd8f2eb
1 parent a869a28
commit bd8f2eb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,