Skip to content

Commit

Permalink
Comment and remove print
Browse files Browse the repository at this point in the history
  • Loading branch information
Swopper050 committed Jan 9, 2024
1 parent e79e9c8 commit 5465c2c
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 3 deletions.
3 changes: 0 additions & 3 deletions ops/opset13/batch_normalization.go
Original file line number Diff line number Diff line change
@@ -1,8 +1,6 @@
package opset13

import (
"fmt"

"github.com/advancedclimatesystems/gonnx/onnx"
"github.com/advancedclimatesystems/gonnx/ops"
"gorgonia.org/tensor"
Expand Down Expand Up @@ -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
Expand Down
2 changes: 2 additions & 0 deletions ops_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -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"
}
Expand Down

0 comments on commit 5465c2c

Please sign in to comment.