Skip to content
This repository has been archived by the owner on Dec 11, 2023. It is now read-only.

Using Pickle with CPU #490

Open
hudamsc2006 opened this issue Sep 27, 2022 · 0 comments
Open

Using Pickle with CPU #490

hudamsc2006 opened this issue Sep 27, 2022 · 0 comments

Comments

@hudamsc2006
Copy link

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)

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant