i tried the above but am still getting a “Given groups=1, weight of size [64, 3, 7, 7], expected input[4, 1, 224, 224] to have 3 channels, but got 1 channels instead” error.
Did you find any other way to go about it
This is my model arch.

Screen Shot 2019-05-16 at 4.11.27 PM.png2376×670 73.8 KB