Skip to content

How to switch NHWC to NCHW #56

New issue
New issue
Open
Open
How to switch NHWC to NCHW#56

Description

@PinkPanther-ny
PinkPanther-ny
opened on Apr 15, 2021

I keep getting this issue and have no progress at all, please help. I've already done a lot of searching on google, Tried reshape the 'x' in model.py predict().
The Conv2D op currently only supports the NHWC tensor format on the CPU. The op was given the format: NCHW

Activity

  1. jooskesters commented on May 10, 2021

    @jooskesters
    jooskesters
    on May 10, 2021

    This is probably because you have the 'wrong' version of tensorflow for this project. For me this worked with tf 2.2.0. Whenever I went higher e.g. 2.4.0, I got the same error as you described.

  2. Cyazd commented on Apr 12, 2022

    @Cyazd
    Cyazd
    on Apr 12, 2022

    TF2.2.0 didn't solve the issue for me. However TF1.6.0 did. All necessary packages are indicated in the requirements.txt files. Just pip install it in a new environment and it should work.

  3. jooskesters commented on Apr 12, 2022

    @jooskesters
    jooskesters
    on Apr 12, 2022 via email
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions