Commit fe895c12 by Javier Ledo Vázquez

Change deprecated api

parent e64e7e5b
......@@ -13,13 +13,13 @@ function depthwiseConvLayer(
return tf.tidy(() => {
let out = tf.depthwiseConv2d(x, params.filters, strides, 'same')
out = tf.batchNormalization<tf.Rank.R4>(
out = tf.batchNorm<tf.Rank.R4>(
out,
params.batch_norm_mean,
params.batch_norm_variance,
epsilon,
params.batch_norm_offset,
params.batch_norm_scale,
params.batch_norm_offset
epsilon
)
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