Using Pickle with CPU

[hudamsc2006]

I have a problem when running my code that trains 2DCNN using pickle, I use CPU not GPU. The error occurs from the the first epoch and the program stop. Kindly can any one help me please.

The error is: UnimplementedError: Exception encountered when calling layer "conv2d_4" " f"(type Conv2D).

{{function_node _wrapped__Conv2D_device/job:localhost/replica:0/task:0/device:CPU:0}} The Conv2D op currently only supports the NHWC tensor format on the CPU. The op was given the format: NCHW [Op:Conv2D]

Call arguments received by layer "conv2d_4" " f"(type Conv2D): • inputs=tf.Tensor(shape=(8, 1, 250, 250), dtype=float32)