"...llama.cpp/git@developer.sourcefind.cn:OpenDAS/ollama.git" did not exist on "cd3fbf1c49aa6e68f982934c3d11b8a4b8e71f58"
Commit 6b5c4dfc authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change

PiperOrigin-RevId: 458476086
parent fd232c2c
# MobileNetV3-large_1.0 ImageNet classification: ~75.3% top-1. # MobileNetV3-large_1.0 ImageNet classification: ~75.7% top-1.
runtime: runtime:
distribution_strategy: 'tpu' distribution_strategy: 'tpu'
mixed_precision_dtype: 'bfloat16' mixed_precision_dtype: 'bfloat16'
...@@ -27,10 +27,10 @@ task: ...@@ -27,10 +27,10 @@ task:
dtype: 'bfloat16' dtype: 'bfloat16'
aug_type: aug_type:
autoaug: autoaug:
augmentation_name: v0 augmentation_name: 'v0'
cutout_const: 100 cutout_const: 100
translate_const: 250 translate_const: 250
type: autoaug type: 'autoaug'
validation_data: validation_data:
input_path: 'imagenet-2012-tfrecord/valid*' input_path: 'imagenet-2012-tfrecord/valid*'
is_training: false is_training: false
...@@ -38,7 +38,7 @@ task: ...@@ -38,7 +38,7 @@ task:
dtype: 'bfloat16' dtype: 'bfloat16'
drop_remainder: false drop_remainder: false
trainer: trainer:
train_steps: 156000 # 500 epochs train_steps: 218000 # 700 epochs
validation_steps: 13 validation_steps: 13
validation_interval: 312 validation_interval: 312
steps_per_loop: 312 # NUM_EXAMPLES (1281167) // global_batch_size steps_per_loop: 312 # NUM_EXAMPLES (1281167) // global_batch_size
...@@ -48,7 +48,7 @@ trainer: ...@@ -48,7 +48,7 @@ trainer:
learning_rate: learning_rate:
cosine: cosine:
alpha: 0.0 alpha: 0.0
decay_steps: 156000 decay_steps: 218000
initial_learning_rate: 0.004 initial_learning_rate: 0.004
name: CosineDecay name: CosineDecay
offset: 0 offset: 0
......
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