This repository was archived by the owner on Dec 11, 2023. It is now read-only.
This repository was archived by the owner on Dec 11, 2023. It is now read-only.
Using Pickle with CPU #490
Description
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)