From 0cda06fa5f161a9b17d4d66b2f523737af57e88a Mon Sep 17 00:00:00 2001 From: Frederick Liu Date: Mon, 11 Jul 2022 12:59:32 -0700 Subject: [PATCH] Internal change PiperOrigin-RevId: 460280458 --- official/projects/detr/experiments/detr_r50_300epochs.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/official/projects/detr/experiments/detr_r50_300epochs.sh b/official/projects/detr/experiments/detr_r50_300epochs.sh index c9f7c84f4..162f97430 100644 --- a/official/projects/detr/experiments/detr_r50_300epochs.sh +++ b/official/projects/detr/experiments/detr_r50_300epochs.sh @@ -3,4 +3,4 @@ python3 official/projects/detr/train.py \ --experiment=detr_coco \ --mode=train_and_eval \ --model_dir=/tmp/logging_dir/ \ - --params_override=task.init_checkpoint='gs://tf_model_garden/vision/resnet50_imagenet/ckpt-62400',trainer.train_steps=554400 + --params_override=task.init_checkpoint='gs://tf_model_garden/vision/resnet50_imagenet/ckpt-62400',trainer.train_steps=554400,trainer.optimizer_config.learning_rate.stepwise.boundaries="[369600]" -- GitLab