diff options
Diffstat (limited to 'training')
| -rw-r--r-- | training/conf/datamodule/transform/lines.yaml | 6 | ||||
| -rw-r--r-- | training/conf/datamodule/transform/paragraphs.yaml | 6 |
2 files changed, 6 insertions, 6 deletions
diff --git a/training/conf/datamodule/transform/lines.yaml b/training/conf/datamodule/transform/lines.yaml index d7e51e1..6bd35ef 100644 --- a/training/conf/datamodule/transform/lines.yaml +++ b/training/conf/datamodule/transform/lines.yaml @@ -9,14 +9,14 @@ color_jitter: random_affine: _target_: torchvision.transforms.RandomAffine degrees: 1 - shear: [-30, 20] + shear: [-10, 10] interpolation: BILINEAR fill: 0 random_perspective: _target_: torchvision.transforms.RandomPerspective - distortion_scale: 0.07 - p: 0.5 + distortion_scale: 0.05 + p: 0.25 fill: 0 to_tensor: diff --git a/training/conf/datamodule/transform/paragraphs.yaml b/training/conf/datamodule/transform/paragraphs.yaml index 5e5314c..4f260d0 100644 --- a/training/conf/datamodule/transform/paragraphs.yaml +++ b/training/conf/datamodule/transform/paragraphs.yaml @@ -17,13 +17,13 @@ color_jitter: random_affine: _target_: torchvision.transforms.RandomAffine degrees: 1 - shear: [-10, 10] + shear: [-5, 5] interpolation: BILINEAR random_perspective: _target_: torchvision.transforms.RandomPerspective - distortion_scale: 0.1 - p: 0.5 + distortion_scale: 0.05 + p: 0.25 fill: 0 to_tensor: |