RuntimeError: Given groups=1, weight of size 32 6 3 3, expected input[8, 3, 336, 336] to have 6 channels, but got 3 channels instead

[mobassir94]

when i try to train using 6 channel i get this error : RuntimeError: Given groups=1, weight of size 32 6 3 3, expected input[8, 3, 336, 336] to have 6 channels, but got 3 channels instead

code is exactly same as this : https://github.com/lyft/nuscenes-devkit/blob/master/notebooks/Reference%20Model.ipynb