How to change keras network to channels_last instead of channels_first?

Viewed 104

What is the code for changing whether the network expects channels to be first or last? I know you can put it in the layer itself, but do I have to write data_format='channels_last' every time? I can somewhat remember there being an easier way, but I've forgotten, and thus my question.

This is a way of doing it, however:

model = tf.keras.Sequential([tf.keras.layers.Conv2D(1,(3,3),data_format="channels_last")])

1 Answers
Related