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
from __future__ import division
from __future__ import print_function
import tensorflow as tf
from tensorflow.python.keras import backend
from tensorflow.python.keras import initializers
from tensorflow.python.keras import layers
from tensorflow.python.keras import models
from tensorflow.python.keras import regularizers
from official.vision.image_classification import imagenet_preprocessing
layers = tf.keras.layers
L2_WEIGHT_DECAY = 1e-4
BATCH_NORM_DECAY = 0.9
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