diff options
| author | aktersnurra <gustaf.rydholm@gmail.com> | 2020-12-07 22:54:04 +0100 |
|---|---|---|
| committer | aktersnurra <gustaf.rydholm@gmail.com> | 2020-12-07 22:54:04 +0100 |
| commit | 25b5d6983d51e0e791b96a76beb7e49f392cd9a8 (patch) | |
| tree | 526ba739714b3d040f7810c1a6be3ff0ba37fdb1 /src/text_recognizer/models/transformer_model.py | |
| parent | 5529e0fc9ca39e81fe0f08a54f257d32f0afe120 (diff) | |
Segmentation working!
Diffstat (limited to 'src/text_recognizer/models/transformer_model.py')
| -rw-r--r-- | src/text_recognizer/models/transformer_model.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/text_recognizer/models/transformer_model.py b/src/text_recognizer/models/transformer_model.py index 968a047..a912122 100644 --- a/src/text_recognizer/models/transformer_model.py +++ b/src/text_recognizer/models/transformer_model.py @@ -18,8 +18,8 @@ class TransformerModel(Model): def __init__( self, - network_fn: Type[nn.Module], - dataset: Type[Dataset], + network_fn: str, + dataset: str, network_args: Optional[Dict] = None, dataset_args: Optional[Dict] = None, metrics: Optional[Dict] = None, |