diff --git a/ops/opset13/batch_normalization.go b/ops/opset13/batch_normalization.go index 8861cf9..c548fb5 100644 --- a/ops/opset13/batch_normalization.go +++ b/ops/opset13/batch_normalization.go @@ -1,8 +1,6 @@ package opset13 import ( - "fmt" - "github.com/advancedclimatesystems/gonnx/onnx" "github.com/advancedclimatesystems/gonnx/ops" "gorgonia.org/tensor" @@ -168,7 +166,6 @@ func (b *BatchNormalization) reshapeTensors(X, scale, bias, mean, variance tenso } func (b *BatchNormalization) testModeCalculation(X, scale, bias, mean, variance tensor.Tensor) (tensor.Tensor, error) { - fmt.Println("joe") newScale, newBias, newMean, newVariance, err := b.reshapeTensors(X, scale, bias, mean, variance) if err != nil { return nil, err diff --git a/ops_test.go b/ops_test.go index a405939..66f4bdf 100644 --- a/ops_test.go +++ b/ops_test.go @@ -195,6 +195,8 @@ func shouldRunTest(folder, opFilter string) bool { } } + // For some reason ONNX decided to not let these testcases match the operator name. + // Here we manually replace the filter with the name ONNX uses for this test case. if opFilter == "test_batchnormalization" { opFilter = "test_batchnorm" }