Skip to content

Commit 4360b99

Browse files
authored
remove potential performance downgrader (#2651)
* Fix transfer learning (#2645) * fix transfer learning * add ParseSingleExample, DecodeBmp tf loader * add corresponding unit tests * remove potential performance downgrader
1 parent 1d5a352 commit 4360b99

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

spark/dl/src/main/scala/com/intel/analytics/bigdl/nn/SpatialBatchNormalization.scala

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -319,7 +319,8 @@ object SpatialBatchNormalization {
319319
val offsetOffset = offset.storageOffset() - 1
320320
var isIdenticalScale = false
321321
var identicalScale = 0.0f
322-
if (scale.stride().length == 1 && scale.stride()(0) == 0 && scaleData.length == 1) {
322+
323+
if (scaleData.length == 1) {
323324
isIdenticalScale = true
324325
identicalScale = scaleData(0)
325326
}

spark/dl/src/test/scala/com/intel/analytics/bigdl/nn/tf/DecodeBmpSpec.scala

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,9 @@ import org.tensorflow.example.Example
2727
class DecodeBmpSerialTest extends ModuleSerializationTest {
2828
private def getInputs(name: String): Tensor[ByteString] = {
2929
import com.intel.analytics.bigdl.utils.tf.TFTensorNumeric.NumericByteString
30+
/* since the tfrecord file is loaded into byteArrays regardless of the
31+
original image type, we can map "bmp" to 0 as well
32+
*/
3033
val index = name match {
3134
case "png" => 0
3235
case "jpeg" => 1

0 commit comments

Comments
 (0)