lib/torch/nn/conv2d.rb in torch-rb-0.1.3 vs lib/torch/nn/conv2d.rb in torch-rb-0.1.4

@@ -17,11 +17,11 @@
         end
         F.conv2d(input, @weight, @bias, stride: @stride, padding: @padding, dilation: @dilation, groups: @groups)
       end
 
       # TODO add more parameters
-      def inspect
-        "Conv2d(#{@in_channels}, #{@out_channels}, kernel_size: #{@kernel_size.inspect}, stride: #{@stride.inspect})"
+      def extra_inspect
+        format("%s, %s, kernel_size: %s, stride: %s", @in_channels, @out_channels, @kernel_size, @stride)
       end
 
       private
 
       def pair(value)