diff --git a/training_args.json b/training_args.json
index c1059b3..001899a 100644
--- a/training_args.json
+++ b/training_args.json
@@ -32,5 +32,6 @@
     "hard_pivot": false,
     "weight_type": "bf16",
     "dataloader_num_workers": 0,
-    "aspect_ratio_bucketing": false
+    "aspect_ratio_bucketing": true,
+    "do_cache": false
 }
\ No newline at end of file