提交 b2539e5e 编写于 作者: J Jiageng Zhang 提交者: A. Unique TensorFlower

Internal change

PiperOrigin-RevId: 525498900
上级 cc7fbe50
......@@ -114,7 +114,7 @@ trainer:
best_checkpoint_eval_metric: AP
best_checkpoint_metric_comp: higher
train_steps: 138600
validation_steps: 19
validation_steps: 312
validation_interval: 2310
steps_per_loop: 462
summary_interval: 462
......
......@@ -70,10 +70,10 @@ task:
train_data:
input_path: /readahead/200M/placer/prod/home/tensorflow-performance-data/datasets/coco/train*
is_training: true
global_batch_size: 64
global_batch_size: 32
dtype: float32
parser:
max_num_instances: 300
max_num_instances: 100
letter_box: true
random_flip: true
random_pad: false
......@@ -101,11 +101,11 @@ task:
validation_data:
input_path: /readahead/200M/placer/prod/home/tensorflow-performance-data/datasets/coco/val*
is_training: false
global_batch_size: 16
global_batch_size: 8
dtype: float32
drop_remainder: true
parser:
max_num_instances: 300
max_num_instances: 100
letter_box: true
use_tie_breaker: true
best_match_only: true
......@@ -115,12 +115,12 @@ trainer:
best_checkpoint_export_subdir: best_ckpt
best_checkpoint_eval_metric: AP
best_checkpoint_metric_comp: higher
train_steps: 554400
validation_steps: 19
validation_interval: 9240
steps_per_loop: 1848
summary_interval: 1848
checkpoint_interval: 1848
train_steps: 1108800
validation_steps: 625
validation_interval: 18480
steps_per_loop: 3696
summary_interval: 3696
checkpoint_interval: 3696
optimizer_config:
ema:
average_decay: 0.9999
......@@ -130,7 +130,7 @@ trainer:
cosine:
initial_learning_rate: 0.01
alpha: 0.1
decay_steps: 554400
decay_steps: 1108800
type: cosine
optimizer:
sgd_torch:
......@@ -139,11 +139,11 @@ trainer:
momentum: 0.937
momentum_start: 0.8
nesterov: true
warmup_steps: 5544
warmup_steps: 11088
weight_decay: 0.0005
type: sgd_torch
warmup:
linear:
warmup_learning_rate: 0.0
warmup_steps: 5544
warmup_steps: 11088
type: linear
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册