The warning shouldn’t have any negative effects, if you are not changing the memory format manually.
You could replace the conv layer with a linear layer, but this should not be necessary and I would recommend to stick to the approach, which makes the code easier to understand.
I think the memory format cannot be automatically detected due to the currently used (ambiguous) kernel shape.