Commit 6c9a23e2 authored by Cristina Vasconcelos's avatar Cristina Vasconcelos Committed by A. Unique TensorFlower
Browse files

Internal change

PiperOrigin-RevId: 470729449
parent 4a8e45d1
task:
# init_checkpoint: 'a_pretrained_backbone_checkpoint'
init_checkpoint_modules: backbone
freeze_backbone: true
model:
backbone:
resnet:
model_id: 101
replace_stem_max_pool: true
resnetd_shortcut: true
scale_stem: true
se_ratio: 0.25
stem_type: v1
type: resnet
decoder:
type: fpn
train_data:
global_batch_size: 256
parser:
aug_rand_hflip: true
aug_scale_max: 2.0
aug_scale_min: 0.1
trainer:
optimizer_config:
learning_rate:
stepwise:
boundaries: [265684, 272615]
name: PiecewiseConstantDecay
offset: 0
values: [0.32, 0.032, 0.0032]
type: stepwise
steps_per_loop: 462
summary_interval: 462
train_steps: 277235
task:
# init_checkpoint: 'a_pretrained_backbone_checkpoint'
init_checkpoint_modules: backbone
freeze_backbone: true
model:
backbone:
resnet:
model_id: 101
replace_stem_max_pool: true
resnetd_shortcut: true
scale_stem: true
se_ratio: 0.25
stem_type: v1
type: resnet
decoder:
type: fpn
train_data:
global_batch_size: 256
parser:
aug_rand_hflip: true
aug_scale_max: 2.0
aug_scale_min: 0.1
trainer:
optimizer_config:
learning_rate:
stepwise:
boundaries: [22176, 31416]
name: PiecewiseConstantDecay
offset: 0
values: [0.32, 0.032, 0.0032]
type: stepwise
steps_per_loop: 462
summary_interval: 462
train_steps: 33264
task:
# init_checkpoint: 'a_pretrained_backbone_checkpoint'
init_checkpoint_modules: backbone
freeze_backbone: true
model:
backbone:
resnet:
model_id: 101
replace_stem_max_pool: true
resnetd_shortcut: true
scale_stem: true
se_ratio: 0.25
stem_type: v1
type: resnet
decoder:
type: nasfpn
train_data:
global_batch_size: 256
parser:
aug_rand_hflip: true
aug_scale_max: 2.0
aug_scale_min: 0.1
trainer:
optimizer_config:
learning_rate:
stepwise:
boundaries: [265684, 272615]
name: PiecewiseConstantDecay
offset: 0
values: [0.32, 0.032, 0.0032]
type: stepwise
steps_per_loop: 462
summary_interval: 462
train_steps: 277235
task:
# init_checkpoint: 'a_pretrained_backbone_checkpoint'
init_checkpoint_modules: backbone
freeze_backbone: true
model:
backbone:
resnet:
model_id: 101
replace_stem_max_pool: true
resnetd_shortcut: true
scale_stem: true
se_ratio: 0.25
stem_type: v1
type: resnet
decoder:
type: nasfpn
train_data:
global_batch_size: 256
parser:
aug_rand_hflip: true
aug_scale_max: 2.0
aug_scale_min: 0.1
trainer:
optimizer_config:
learning_rate:
stepwise:
boundaries: [22176, 31416]
name: PiecewiseConstantDecay
offset: 0
values: [0.32, 0.032, 0.0032]
type: stepwise
steps_per_loop: 462
summary_interval: 462
train_steps: 33264
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment