--inv-disc=False throws an error

[Oliver Thomas]

throws
`---------------------------------------------------------------------------
RuntimeError Traceback (most recent call last)
in ()
----> 1 run.main(["--inv-disc=False", "--epochs=2", "--data-pcnt=0.01"])

11 frames
/usr/local/lib/python3.6/dist-packages/torch/nn/modules/conv.py in forward(self, input)
336 _pair(0), self.dilation, self.groups)
337 return F.conv2d(input, self.weight, self.bias, self.stride,
--> 338 self.padding, self.dilation, self.groups)
339
340

RuntimeError: Given groups=1, weight of size 20 1 5 5, expected input[100, 3, 28, 28] to have 1 channels, but got 3 channels instead`

[Oliver Thomas]

Have run with this flag. it runs. @MylesBartlett advises that he fixed it