summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-02-11 22:51:31 +0100
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-02-11 22:51:31 +0100
commit94ec29dc00b8e19dd7fcdaac0856b6cf00abd469 (patch)
treedc7d08f997bbde1d5d3f58051d7b6bce006738e8
parentadc36f7892ffedeafed7579282d62fbeecd496e9 (diff)
fix(paragraphs-conf): replace random crop with center crop
-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