Skip to content

Commit

Permalink
update with latest changes
Browse files Browse the repository at this point in the history
  • Loading branch information
felixdittrich92 committed Sep 15, 2022
1 parent 16a5b0f commit ee253d0
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion doctr/models/recognition/vitstr/tensorflow.py
Original file line number Diff line number Diff line change
Expand Up @@ -194,7 +194,7 @@ def _vitstr(
kwargs["vocab"] = _cfg["vocab"]

# Feature extractor
feat_extractor = vit_b(
feat_extractor = vit_b( # type: ignore[operator]
pretrained=pretrained_backbone,
input_shape=_cfg["input_shape"],
patch_size=(4, 8),
Expand Down

0 comments on commit ee253d0

Please sign in to comment.