Skip to content
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

@hudamsc2006

Description

@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)

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions