Skip to content

Commit c182f41

Browse files
author
Alejandro Gaston Alvarez Franceschi
committed
Remove eval
1 parent 95bf8ea commit c182f41

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

coremltools/converters/mil/frontend/torch/test/test_torch_ops.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1715,7 +1715,7 @@ def forward(self, input_data, weights):
17151715
input_data, weights, stride=self.stride, padding=self.padding, groups=self.groups
17161716
)
17171717

1718-
model = FunctionalConv1D().eval()
1718+
model = FunctionalConv1D()
17191719
input_shape = [
17201720
(1, in_channels, width),
17211721
(out_channels, int(in_channels / groups), kernel_size),
@@ -1790,7 +1790,7 @@ def forward(self, input_data, weights):
17901790
input_data, weights, stride=self.stride, padding=self.padding, groups=self.groups
17911791
)
17921792

1793-
model = FunctionalConv2D().eval()
1793+
model = FunctionalConv2D()
17941794

17951795
input_shape = [
17961796
(1, in_channels, height, width),
@@ -1868,7 +1868,7 @@ def forward(self, input_data, weights):
18681868
input_data, weights, stride=self.stride, padding=self.padding, groups=self.groups
18691869
)
18701870

1871-
model = FunctionalConv3D().eval()
1871+
model = FunctionalConv3D()
18721872
input_shape = [
18731873
(1, in_channels, depth, height, width),
18741874
(out_channels, int(in_channels / groups), kernel_size, kernel_size, kernel_size),

0 commit comments

Comments
 (0)