diff --git a/text_to_image_reference_using_torch_loadgen/data_axs.json b/text_to_image_reference_using_torch_loadgen/data_axs.json
index a22a4d6..139e04a 100644
--- a/text_to_image_reference_using_torch_loadgen/data_axs.json
+++ b/text_to_image_reference_using_torch_loadgen/data_axs.json
@@ -89,8 +89,8 @@
         [ "get_path" ]
     ]] ],
 
-    "loadgen_dataset_size": 10,
-    "loadgen_buffer_size": 1,
+    "loadgen_dataset_size": 5000,
+    "loadgen_buffer_size": 5000,
 
     "device": [ "^", "case",[ ["^^", "get", "num_gpus"], "0", "cpu" ], {"default_value": "cuda"}, ["num_gpus"]],
     "dtype": "fp32",