Fix DP-SGD model caching directories.

google · Oct 12, 2024 · a87da99 · a87da99
1 parent ce9f786
commit a87da99
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tf_shell_ml/test/dpsgd_model_distrib_test.py b/tf_shell_ml/test/dpsgd_model_distrib_test.py
@@ -79,7 +79,7 @@ def test_model(self):
         val_dataset = tf.data.Dataset.from_tensor_slices((x_test, y_test))
         val_dataset = val_dataset.batch(32)
 
-        context_cache_path = "/tmp/postscale_model_distrib_test_cache/"
+        context_cache_path = "/tmp/dpsgd_model_distrib_test_cache/"
         os.makedirs(context_cache_path, exist_ok=True)
 
         m = tf_shell_ml.DpSgdSequential(

diff --git a/tf_shell_ml/test/dpsgd_model_local_test.py b/tf_shell_ml/test/dpsgd_model_local_test.py
@@ -40,7 +40,7 @@ def _test_model(self, disable_encryption, disable_masking, disable_noise):
         val_dataset = tf.data.Dataset.from_tensor_slices((x_test, y_test))
         val_dataset = val_dataset.batch(32)
 
-        context_cache_path = "/tmp/postscale_model_local_test_cache/"
+        context_cache_path = "/tmp/dpsgd_model_local_test_cache/"
         os.makedirs(context_cache_path, exist_ok=True)
 
         # Turn on the shell optimizer to use autocontext.