Remove eval

apple · Jan 4, 2024 · c182f41 · c182f41
1 parent 95bf8ea
commit c182f41
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/coremltools/converters/mil/frontend/torch/test/test_torch_ops.py b/coremltools/converters/mil/frontend/torch/test/test_torch_ops.py
@@ -1715,7 +1715,7 @@ def forward(self, input_data, weights):
                     input_data, weights, stride=self.stride, padding=self.padding, groups=self.groups
                 )
 
-        model = FunctionalConv1D().eval()
+        model = FunctionalConv1D()
         input_shape = [
             (1, in_channels, width),
             (out_channels, int(in_channels / groups), kernel_size),
@@ -1790,7 +1790,7 @@ def forward(self, input_data, weights):
                     input_data, weights, stride=self.stride, padding=self.padding, groups=self.groups
                 )
 
-        model = FunctionalConv2D().eval()
+        model = FunctionalConv2D()
 
         input_shape = [
             (1, in_channels, height, width),
@@ -1868,7 +1868,7 @@ def forward(self, input_data, weights):
                     input_data, weights, stride=self.stride, padding=self.padding, groups=self.groups
                 )
 
-        model = FunctionalConv3D().eval()
+        model = FunctionalConv3D()
         input_shape = [
             (1, in_channels, depth, height, width),
             (out_channels, int(in_channels / groups), kernel_size, kernel_size, kernel_size),