diff --git a/official/projects/detr/experiments/detr_r50_300epochs.sh b/official/projects/detr/experiments/detr_r50_300epochs.sh index c9f7c84f4b13a82b2d9e57cdc56dc136efeddeb5..162f974306bfdaed79cf0f092285bb614ec7fc69 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]"