summaryrefslogtreecommitdiff
path: root/training/conf
diff options
context:
space:
mode:
Diffstat (limited to 'training/conf')
-rw-r--r--training/conf/datamodule/transform/lines.yaml6
-rw-r--r--training/conf/datamodule/transform/paragraphs.yaml6
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: