Update hf-diffusers.libsonnet (GoogleCloudPlatform#1039)

Copy from GoogleCloudPlatform#1038 to r2.2
RissyRan · Jan 9, 2024 · a9260e9 · a9260e9
1 parent 5aed632
commit a9260e9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/pytorch/r2.2/hf-diffusers.libsonnet b/tests/pytorch/r2.2/hf-diffusers.libsonnet
@@ -22,7 +22,8 @@ local tpus = import 'templates/tpus.libsonnet';
       '--lr_scheduler=constant',
       '--lr_warmup_steps=0',
       '--output_dir=/tmp/sd-pokemon-model',
-      '--checkpoints_total_limit=3',
+      '--checkpoints_total_limit=1',
+      '--checkpointing_steps=6000',
     ],
   },