From 95cbdf5bc1cc9639febda23c28d8f464c998b214 Mon Sep 17 00:00:00 2001
From: aktersnurra <gustaf.rydholm@gmail.com>
Date: Tue, 11 Aug 2020 23:08:56 +0200
Subject: Working one the cnn lstm ctc model.

---
 src/text_recognizer/datasets/util.py | 1 +
 1 file changed, 1 insertion(+)

(limited to 'src/text_recognizer/datasets')

diff --git a/src/text_recognizer/datasets/util.py b/src/text_recognizer/datasets/util.py
index 321bc67..76bd85f 100644
--- a/src/text_recognizer/datasets/util.py
+++ b/src/text_recognizer/datasets/util.py
@@ -46,6 +46,7 @@ def fetch_data_loaders(
     """
 
     def check_dataset_args(args: Dict, split: str) -> Dict:
+        """Adds train flag to the dataset args."""
         args["train"] = True if split == "train" else False
         return args
 
-- 
cgit v1.2.3-70-g09d2