Skip to content

GitLab

Commit 8d5c1684 authored by Toby Boyd's avatar Toby Boyd
Browse files

decay changed 2e-4 and batch_norm_decay .997 

 - Average of 5 runs single gpu is 93%+ range 92.5 to 93.4 top_1
parent b83f1e88
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/resnet/keras/resnet_cifar_model.py
View file @ 8d5c1684
...@@ -28,9 +28,9 @@ import warnings ...@@ -28,9 +28,9 @@ import warnings
import tensorflow as tf import tensorflow as tf
BATCH_NORM_DECAY = 0.9 BATCH_NORM_DECAY = 0.997
BATCH_NORM_EPSILON = 1e-5 BATCH_NORM_EPSILON = 1e-5
L2_WEIGHT_DECAY = 1e-4 L2_WEIGHT_DECAY = 2e-4
def _obtain_input_shape(input_shape, def _obtain_input_shape(input_shape,
......
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