Reducing batch size on blueprint to avoid OOMs (#492) (#494)

gretelai · Dec 5, 2024 · 0902097 · 0902097
1 parent a9cbe0a
commit 0902097
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/config_templates/gretel/synthetics/navigator-ft-differential-privacy.yml b/config_templates/gretel/synthetics/navigator-ft-differential-privacy.yml
@@ -27,7 +27,7 @@ models:
         num_input_records_to_sample: auto
 
         # You can try increasing this until you run out-of-memory.
-        batch_size: 8
+        batch_size: 4
 
     privacy_params:
         dp: true

diff --git a/use_cases/gretel.json b/use_cases/gretel.json
@@ -98,7 +98,7 @@
         "description": "This dataset contains reviews, ratings, and categories for women's clothing.",
         "records": 20641,
         "fields": 8,
-        "trainingTime": "39 mins",
+        "trainingTime": "46 mins",
         "bytes": 3260007
       }
     },