Skip to content

GitLab

Commit caa5158f authored by Jaehong Kim's avatar Jaehong Kim Committed by A. Unique TensorFlower
Browse files

Enable v2 layers for resnet_model. 

PiperOrigin-RevId: 286171892
parent b37c3fc1
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 1 deletion
official/vision/image_classification/resnet_model.py
View file @ caa5158f
...@@ -27,13 +27,16 @@ from __future__ import absolute_import ...@@ -27,13 +27,16 @@ from __future__ import absolute_import
from __future__ import division from __future__ import division
from __future__ import print_function from __future__ import print_function
import tensorflow as tf
from tensorflow.python.keras import backend from tensorflow.python.keras import backend
from tensorflow.python.keras import initializers from tensorflow.python.keras import initializers
from tensorflow.python.keras import layers
from tensorflow.python.keras import models from tensorflow.python.keras import models
from tensorflow.python.keras import regularizers from tensorflow.python.keras import regularizers
from official.vision.image_classification import imagenet_preprocessing from official.vision.image_classification import imagenet_preprocessing
layers = tf.keras.layers
L2_WEIGHT_DECAY = 1e-4 L2_WEIGHT_DECAY = 1e-4
BATCH_NORM_DECAY = 0.9 BATCH_NORM_DECAY = 0.9
BATCH_NORM_EPSILON = 1e-5 BATCH_NORM_EPSILON = 1e-5
......
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