Skip to content

GitLab

Commit 3dee3b86 authored by Alan Chiao's avatar Alan Chiao Committed by A. Unique TensorFlower
Browse files

Enable v2 layers for resnet_model. 

PiperOrigin-RevId: 287370879
parent c413c90e
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 4 deletions
official/vision/image_classification/resnet_model.py
View file @ 3dee3b86
...@@ -27,16 +27,13 @@ from __future__ import absolute_import ...@@ -27,16 +27,13 @@ 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