disable pretrained backbone by default (#1080)

mindee · Sep 29, 2022 · dcbb21f · dcbb21f
1 parent 147b60b
commit dcbb21f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/doctr/models/recognition/vitstr/pytorch.py b/doctr/models/recognition/vitstr/pytorch.py
@@ -182,7 +182,7 @@ def _vitstr(
     pretrained: bool,
     backbone_fn: Callable[[bool], nn.Module],
     layer: str,
-    pretrained_backbone: bool = True,
+    pretrained_backbone: bool = False,  # NOTE: training from scratch without a pretrained backbone works better
     ignore_keys: Optional[List[str]] = None,
     **kwargs: Any,
 ) -> ViTSTR:

diff --git a/doctr/models/recognition/vitstr/tensorflow.py b/doctr/models/recognition/vitstr/tensorflow.py
@@ -186,7 +186,7 @@ def _vitstr(
     arch: str,
     pretrained: bool,
     backbone_fn,
-    pretrained_backbone: bool = True,
+    pretrained_backbone: bool = False,  # NOTE: training from scratch without a pretrained backbone works better
     input_shape: Optional[Tuple[int, int, int]] = None,
     **kwargs: Any,
 ) -> ViTSTR: