Skip to content

F
face
Commit fe895c12 by Javier Ledo Vázquez
Options

Change deprecated api

parent e64e7e5b
Showing with 3 additions and 3 deletions
src/ssdMobilenetv1/mobileNetV1.ts
...@@ -13,13 +13,13 @@ function depthwiseConvLayer( ...@@ -13,13 +13,13 @@ function depthwiseConvLayer(
return tf.tidy(() => { return tf.tidy(() => {
let out = tf.depthwiseConv2d(x, params.filters, strides, 'same') let out = tf.depthwiseConv2d(x, params.filters, strides, 'same')
out = tf.batchNormalization<tf.Rank.R4>( out = tf.batchNorm<tf.Rank.R4>(
out, out,
params.batch_norm_mean, params.batch_norm_mean,
params.batch_norm_variance, params.batch_norm_variance,
epsilon, params.batch_norm_offset,
params.batch_norm_scale, params.batch_norm_scale,
params.batch_norm_offset epsilon
) )
return tf.clipByValue(out, 0, 6) return tf.clipByValue(out, 0, 6)
......
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