Updated loadgen_buffer_size and loadgen_dataset_size for text_to_imag…

…e_reference_using_torch_loadgen
krai · Jan 23, 2024 · 9c8c0af · 9c8c0af
1 parent 266676c
commit 9c8c0af
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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",