diff options
Diffstat (limited to 'training')
| -rw-r--r-- | training/conf/datamodule/iam_lines.yaml | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/training/conf/datamodule/iam_lines.yaml b/training/conf/datamodule/iam_lines.yaml index ce4fa91..19c72f8 100644 --- a/training/conf/datamodule/iam_lines.yaml +++ b/training/conf/datamodule/iam_lines.yaml @@ -2,6 +2,6 @@ _target_: text_recognizer.data.iam_lines.IAMLines batch_size: 8 num_workers: 12 train_fraction: 0.8 -pin_memory: false +pin_memory: true transform: transform/lines.yaml test_transform: test_transform/lines.yaml |