summaryrefslogtreecommitdiff
path: root/training/conf/datamodule
diff options
context:
space:
mode:
Diffstat (limited to 'training/conf/datamodule')
-rw-r--r--training/conf/datamodule/transform/paragraphs.yaml16
1 files changed, 10 insertions, 6 deletions
diff --git a/training/conf/datamodule/transform/paragraphs.yaml b/training/conf/datamodule/transform/paragraphs.yaml
index 1b0ab04..5e5314c 100644
--- a/training/conf/datamodule/transform/paragraphs.yaml
+++ b/training/conf/datamodule/transform/paragraphs.yaml
@@ -1,10 +1,14 @@
-random_crop:
- _target_: torchvision.transforms.RandomCrop
+# random_crop:
+# _target_: torchvision.transforms.RandomCrop
+# size: [576, 640]
+# padding: null
+# pad_if_needed: true
+# fill: 0
+# padding_mode: constant
+
+center_crop:
+ _target_: torchvision.transforms.CenterCrop
size: [576, 640]
- padding: null
- pad_if_needed: true
- fill: 0
- padding_mode: constant
color_jitter:
_target_: torchvision.transforms.ColorJitter