Change deprecated api

javyxx · javyxx · commit fe895c125485 · 2019-03-17T21:16:46.000+01:00
diff --git a/src/ssdMobilenetv1/mobileNetV1.ts b/src/ssdMobilenetv1/mobileNetV1.ts
@@ -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)
 
