tensorflow / nmt

TensorFlow Neural Machine Translation Tutorial
Apache License 2.0
6.35k stars 1.96k forks source link

Using Pickle with CPU #490

Open hudamsc2006 opened 1 year ago

hudamsc2006 commented 1 year ago

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 wrappedConv2Ddevice/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)