disable random resize crop + ft config

2026-05-16 11:04:00 +02:00
parent 06f0c3fa78
commit 4df162180e
5 changed files with 63 additions and 2 deletions
@@ -78,6 +78,7 @@ def main(args, resume_preempt=False):
    use_horizontal_flip = args["data"]["use_horizontal_flip"]
    use_color_distortion = args["data"]["use_color_distortion"]
    color_jitter = args["data"]["color_jitter_strength"]
+    use_random_resized_crop = args["data"].get("use_random_resized_crop", True)
    # --
    batch_size = args["data"]["batch_size"]
    pin_mem = args["data"]["pin_mem"]
@@ -181,6 +182,7 @@ def main(args, resume_preempt=False):
        horizontal_flip=use_horizontal_flip,
        color_distortion=use_color_distortion,
        color_jitter=color_jitter,
+        use_random_resized_crop=use_random_resized_crop,
    )

    # -- init data-loaders/samplers