lib/torch/nn/conv2d.rb in torch-rb-0.2.7 vs lib/torch/nn/conv2d.rb in torch-rb-0.3.0

@@ -16,10 +16,9 @@
           raise NotImplementedError
         end
         F.conv2d(input, @weight, @bias, @stride, @padding, @dilation, @groups)
       end
 
-      # TODO add more parameters
       def extra_inspect
         s = String.new("%{in_channels}, %{out_channels}, kernel_size: %{kernel_size}, stride: %{stride}")
         s += ", padding: %{padding}" if @padding != [0] * @padding.size
         s += ", dilation: %{dilation}" if @dilation != [1] * @dilation.size
         s += ", output_padding: %{output_padding}" if @output_padding != [0] * @output_padding.size