Added info about default value to docstring

Signed-off-by: Jim Martens <github@2martens.de>
2019-04-04 17:30:32 +02:00
parent 05fa489905
commit d267512312
1 changed files with 1 additions and 1 deletions
--- a/src/twomartens/masterthesis/data.py
+++ b/src/twomartens/masterthesis/data.py
@ -34,7 +34,7 @@ def load_coco(data_path: str, data_type: str, num_epochs: int, batch_size: int =
        data_path: path to the COCO data set
        data_type: type of the COCO data (e.g. 'val2014')
        num_epochs: number of epochs
-        batch_size: batch size
+        batch_size: batch size (default: 32)
    Returns:
        Tensorflow data set
    """