From afe10f3bed89a30ae77404846a2c1483b20e3105 Mon Sep 17 00:00:00 2001 From: Pei Zhang Date: Tue, 9 Jan 2024 14:38:01 -0800 Subject: [PATCH] Update hf-diffusers.libsonnet Copy from https://github.com/GoogleCloudPlatform/ml-testing-accelerators/pull/1038 to r2.2 --- tests/pytorch/r2.2/hf-diffusers.libsonnet | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/pytorch/r2.2/hf-diffusers.libsonnet b/tests/pytorch/r2.2/hf-diffusers.libsonnet index 35e5e9400..385447183 100644 --- 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', ], },